1 #include <gtest/gtest.h>
2
3 #include <c10/core/Scalar.h>
4
5 using namespace c10;
6
TEST(ScalarTest,UnsignedConstructor)7 TEST(ScalarTest, UnsignedConstructor) {
8 uint16_t x = 0xFFFF;
9 uint32_t y = 0xFFFFFFFF;
10 uint64_t z0 = 0;
11 uint64_t z1 = 0x7FFFFFFFFFFFFFFF;
12 uint64_t z2 = 0xFFFFFFFFFFFFFFFF;
13 auto sx = Scalar(x);
14 auto sy = Scalar(y);
15 auto sz0 = Scalar(z0);
16 auto sz1 = Scalar(z1);
17 auto sz2 = Scalar(z2);
18 ASSERT_TRUE(sx.isIntegral(false));
19 ASSERT_TRUE(sy.isIntegral(false));
20 ASSERT_TRUE(sz0.isIntegral(false));
21 ASSERT_TRUE(sz1.isIntegral(false));
22 ASSERT_TRUE(sz2.isIntegral(false));
23 ASSERT_EQ(sx.type(), ScalarType::Long);
24 ASSERT_EQ(sy.type(), ScalarType::Long);
25 ASSERT_EQ(sz0.type(), ScalarType::Long);
26 ASSERT_EQ(sz1.type(), ScalarType::Long);
27 ASSERT_EQ(sz2.type(), ScalarType::UInt64);
28 ASSERT_EQ(sx.toUInt16(), x);
29 ASSERT_EQ(sx.toInt(), x);
30 ASSERT_EQ(sy.toUInt32(), y);
31 EXPECT_THROW(sy.toInt(), std::runtime_error); // overflows
32 ASSERT_EQ(sy.toLong(), y);
33 ASSERT_EQ(sz0.toUInt64(), z0);
34 ASSERT_EQ(sz0.toInt(), z0);
35 ASSERT_EQ(sz1.toUInt64(), z1);
36 EXPECT_THROW(sz1.toInt(), std::runtime_error); // overflows
37 ASSERT_EQ(sz1.toLong(), z1);
38 ASSERT_EQ(sz2.toUInt64(), z2);
39 EXPECT_THROW(sz2.toInt(), std::runtime_error); // overflows
40 EXPECT_THROW(sz2.toLong(), std::runtime_error); // overflows
41 }
42
TEST(ScalarTest,Equality)43 TEST(ScalarTest, Equality) {
44 ASSERT_TRUE(Scalar(static_cast<uint64_t>(0xFFFFFFFFFFFFFFFF))
45 .equal(0xFFFFFFFFFFFFFFFF));
46 ASSERT_FALSE(Scalar(0).equal(0xFFFFFFFFFFFFFFFF));
47 // ensure that we don't incorrectly coerce bitrep
48 ASSERT_FALSE(Scalar(-1).equal(0xFFFFFFFFFFFFFFFF));
49 }
50
TEST(ScalarTest,LongsAndLongLongs)51 TEST(ScalarTest, LongsAndLongLongs) {
52 Scalar longOne = 1L;
53 Scalar longlongOne = 1LL;
54 ASSERT_EQ(longOne.toInt(), longlongOne.toInt());
55 }
56