shuffle_ut.cpp 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. #include "fast.h"
  2. #include "shuffle.h"
  3. #include "mersenne.h"
  4. #include <library/cpp/testing/unittest/registar.h>
  5. #include <util/generic/ylimits.h>
  6. Y_UNIT_TEST_SUITE(TRandUtilsTest) {
  7. template <typename... A>
  8. static void TestRange(A&&... args) {
  9. TString s0, s1;
  10. ShuffleRange(s1, args...);
  11. s1 = "0";
  12. ShuffleRange(s1, args...);
  13. s1 = "01";
  14. ShuffleRange(s1, args...);
  15. s1 = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
  16. s0 = s1.copy();
  17. ShuffleRange(s1, args...);
  18. UNIT_ASSERT(s0 != s1); // if shuffle does work, chances it will fail are 1 to 64!.
  19. }
  20. template <typename... A>
  21. static void TestIter(A&&... args) {
  22. TString s0, s1;
  23. auto f = [&]() {
  24. auto b = s1.begin();
  25. auto e = s1.end();
  26. Shuffle(b, e, args...);
  27. };
  28. s1 = "0";
  29. f();
  30. s1 = "01";
  31. f();
  32. s1 = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
  33. s0 = s1.copy();
  34. f();
  35. UNIT_ASSERT(s0 != s1); // if shuffle does work, chances it will fail are 1 to 64!.
  36. }
  37. Y_UNIT_TEST(TestShuffle) {
  38. TestRange();
  39. }
  40. Y_UNIT_TEST(TestShuffleMersenne64) {
  41. TMersenne<ui64> prng(42);
  42. TestRange(prng);
  43. }
  44. Y_UNIT_TEST(TestShuffleMersenne32) {
  45. TMersenne<ui32> prng(24);
  46. TestIter(prng);
  47. }
  48. Y_UNIT_TEST(TestShuffleFast32) {
  49. TFastRng32 prng(24, 0);
  50. TestIter(prng);
  51. }
  52. Y_UNIT_TEST(TestShuffleFast64) {
  53. TFastRng64 prng(24, 0, 25, 1);
  54. TestIter(prng);
  55. }
  56. }