diff options
Diffstat (limited to 'absl/random/mock_distributions_test.cc')
-rw-r--r-- | absl/random/mock_distributions_test.cc | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/absl/random/mock_distributions_test.cc b/absl/random/mock_distributions_test.cc index 917799f0..80df8714 100644 --- a/absl/random/mock_distributions_test.cc +++ b/absl/random/mock_distributions_test.cc @@ -19,6 +19,7 @@ #include "gmock/gmock.h" #include "gtest/gtest.h" +#include "absl/numeric/int128.h" #include "absl/random/distributions.h" #include "absl/random/mocking_bit_gen.h" #include "absl/random/random.h" @@ -81,6 +82,14 @@ TEST(MockUniform, OutOfBoundsIsAllowed) { EXPECT_EQ(absl::Uniform<int>(gen, 1, 100), 0); } +TEST(ValidatedMockDistributions, UniformUInt128Works) { + absl::random_internal::MockingBitGenImpl<true> gen; + + EXPECT_CALL(absl::MockUniform<absl::uint128>(), Call(gen)) + .WillOnce(Return(absl::Uint128Max())); + EXPECT_EQ(absl::Uniform<absl::uint128>(gen), absl::Uint128Max()); +} + TEST(ValidatedMockDistributions, UniformDoubleBoundaryCases) { absl::random_internal::MockingBitGenImpl<true> gen; |