1 // Copyright 2018 The Abseil Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "absl/random/mock_distributions.h"
16
17 #include "gtest/gtest.h"
18 #include "absl/random/mocking_bit_gen.h"
19 #include "absl/random/random.h"
20
21 namespace {
22 using ::testing::Return;
23
TEST(MockDistributions,Examples)24 TEST(MockDistributions, Examples) {
25 absl::MockingBitGen gen;
26
27 EXPECT_NE(absl::Uniform<int>(gen, 1, 1000000), 20);
28 EXPECT_CALL(absl::MockUniform<int>(), Call(gen, 1, 1000000))
29 .WillOnce(Return(20));
30 EXPECT_EQ(absl::Uniform<int>(gen, 1, 1000000), 20);
31
32 EXPECT_NE(absl::Uniform<double>(gen, 0.0, 100.0), 5.0);
33 EXPECT_CALL(absl::MockUniform<double>(), Call(gen, 0.0, 100.0))
34 .WillOnce(Return(5.0));
35 EXPECT_EQ(absl::Uniform<double>(gen, 0.0, 100.0), 5.0);
36
37 EXPECT_NE(absl::Exponential<double>(gen, 1.0), 42);
38 EXPECT_CALL(absl::MockExponential<double>(), Call(gen, 1.0))
39 .WillOnce(Return(42));
40 EXPECT_EQ(absl::Exponential<double>(gen, 1.0), 42);
41
42 EXPECT_NE(absl::Poisson<int>(gen, 1.0), 500);
43 EXPECT_CALL(absl::MockPoisson<int>(), Call(gen, 1.0)).WillOnce(Return(500));
44 EXPECT_EQ(absl::Poisson<int>(gen, 1.0), 500);
45
46 EXPECT_NE(absl::Bernoulli(gen, 0.000001), true);
47 EXPECT_CALL(absl::MockBernoulli(), Call(gen, 0.000001))
48 .WillOnce(Return(true));
49 EXPECT_EQ(absl::Bernoulli(gen, 0.000001), true);
50
51 EXPECT_NE(absl::Beta<double>(gen, 3.0, 2.0), 0.567);
52 EXPECT_CALL(absl::MockBeta<double>(), Call(gen, 3.0, 2.0))
53 .WillOnce(Return(0.567));
54 EXPECT_EQ(absl::Beta<double>(gen, 3.0, 2.0), 0.567);
55
56 EXPECT_NE(absl::Zipf<int>(gen, 1000000, 2.0, 1.0), 1221);
57 EXPECT_CALL(absl::MockZipf<int>(), Call(gen, 1000000, 2.0, 1.0))
58 .WillOnce(Return(1221));
59 EXPECT_EQ(absl::Zipf<int>(gen, 1000000, 2.0, 1.0), 1221);
60
61 EXPECT_NE(absl::Gaussian<double>(gen, 0.0, 1.0), 0.001);
62 EXPECT_CALL(absl::MockGaussian<double>(), Call(gen, 0.0, 1.0))
63 .WillOnce(Return(0.001));
64 EXPECT_EQ(absl::Gaussian<double>(gen, 0.0, 1.0), 0.001);
65
66 EXPECT_NE(absl::LogUniform<int>(gen, 0, 1000000, 2), 2040);
67 EXPECT_CALL(absl::MockLogUniform<int>(), Call(gen, 0, 1000000, 2))
68 .WillOnce(Return(2040));
69 EXPECT_EQ(absl::LogUniform<int>(gen, 0, 1000000, 2), 2040);
70 }
71
72 } // namespace
73