llvm/libcxx/test/std/numerics/rand/rand.dist/rand.dist.uni/rand.dist.uni.int/int128.pass.cpp

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// UNSUPPORTED: c++03

// <random>

// template<class _IntType = int>
// class uniform_int_distribution

// template<class _URNG> result_type operator()(_URNG& g);


#include <cassert>
#include <cstdint>
#include <limits>
#include <random>

#include "test_macros.h"

int main(int, char**) {

#ifndef TEST_HAS_NO_INT128

  // Test that values outside of the 64-bit range can be produced.
  {
    std::minstd_rand0 e;
    std::uniform_int_distribution<__int128_t> d;
    assert(d.min() == 0 && d.max() == std::numeric_limits<__int128_t>::max());
    bool all_in_64bit_range = true;
    for (int i = 0; i < 100; ++i) {
      __int128_t n = d(e);
      all_in_64bit_range = all_in_64bit_range && (n <= UINT64_MAX);
    }
    assert(!all_in_64bit_range);
  }

  // Same test as above with min/max set and outside the 64-bit range.
  {
    __int128_t a = ((__int128_t)INT64_MIN) * 10;
    __int128_t b = ((__int128_t)INT64_MAX) * 10;
    std::minstd_rand0 e;
    std::uniform_int_distribution<__int128_t> d(a, b);
    assert(d.min() == a && d.max() == b);
    bool all_in_64bit_range = true;
    for (int i = 0; i < 100; ++i) {
      __int128_t n = d(e);
      assert(a <= n && n <= b);
      all_in_64bit_range = all_in_64bit_range && (INT64_MIN <= n) && (n <= (INT64_MAX));
    }
    assert(!all_in_64bit_range);
  }

  // Same test as above with __uint128_t.
  {
    __uint128_t a = UINT64_MAX / 3;
    __uint128_t b = ((__uint128_t)UINT64_MAX) * 10;
    std::minstd_rand0 e;
    std::uniform_int_distribution<__uint128_t> d(a, b);
    assert(d.min() == a && d.max() == b);
    bool all_in_64bit_range = true;
    for (int i = 0; i < 100; ++i) {
      __uint128_t n = d(e);
      assert(a <= n && n <= b);
      all_in_64bit_range = all_in_64bit_range && (n <= (UINT64_MAX));
    }
    assert(!all_in_64bit_range);
  }

  // Regression test for PR#51520:
  {
    std::minstd_rand0 e;
    std::uniform_int_distribution<__int128_t> d(INT64_MIN, INT64_MAX);
    assert(d.min() == INT64_MIN && d.max() == INT64_MAX);
    for (int i = 0; i < 100; ++i) {
      __int128_t n = d(e);
      assert((INT64_MIN <= n) && (n <= INT64_MAX));
    }
  }

#endif // TEST_HAS_NO_INT128

  return 0;
}