diff options
author | Android Build Coastguard Worker <android-build-coastguard-worker@google.com> | 2022-09-14 05:38:41 +0000 |
---|---|---|
committer | Android Build Coastguard Worker <android-build-coastguard-worker@google.com> | 2022-09-14 05:38:41 +0000 |
commit | df9234b8dfd77f79aa64f81af44aa532c9de0a0f (patch) | |
tree | 101bfc731fc453160421993180e07d015e40bc56 /native/utils/testing/test_data_generator.h | |
parent | 15d8bd33b54a5beec49dd759686341ee05e4e109 (diff) | |
parent | 1075b1e4e39ab4af90deb3758e5631943c07d47e (diff) | |
download | libtextclassifier-df9234b8dfd77f79aa64f81af44aa532c9de0a0f.tar.gz |
Snap for 9061588 from 1075b1e4e39ab4af90deb3758e5631943c07d47e to mainline-permission-releaseaml_per_331115020
Change-Id: Iaaf56083744c9820adbab66355934e389b2145e1
Diffstat (limited to 'native/utils/testing/test_data_generator.h')
-rw-r--r-- | native/utils/testing/test_data_generator.h | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/native/utils/testing/test_data_generator.h b/native/utils/testing/test_data_generator.h index c23b5dc..c2d3fff 100644 --- a/native/utils/testing/test_data_generator.h +++ b/native/utils/testing/test_data_generator.h @@ -19,8 +19,10 @@ #include <algorithm> #include <iostream> +#include <limits> #include <random> #include <string> +#include <type_traits> #include "utils/strings/stringpiece.h" @@ -32,8 +34,11 @@ class TestDataGenerator { template <typename T, typename std::enable_if_t<std::is_integral<T>::value>* = nullptr> T generate() { - std::uniform_int_distribution<T> dist; - return dist(random_engine_); + typedef typename std::conditional<sizeof(T) >= sizeof(int16_t), T, + std::int16_t>::type rand_type; + std::uniform_int_distribution<rand_type> dist( + std::numeric_limits<T>::min(), std::numeric_limits<T>::max()); + return static_cast<T>(dist(random_engine_)); } template <> |