@@ -264,16 +264,18 @@ void TestArray2() {
264
264
Array1<T> arr1 (context, data);
265
265
Array2<T> array (arr1, kDim0 , kDim1 );
266
266
Array2<T> array_copy = array.Generic ().template Specialize <T>();
267
- K2_CHECK_EQ ( true , Equal (array, array_copy));
268
- K2_CHECK_EQ (array.Data (), array_copy.Data ());
267
+ EXPECT_TRUE ( Equal (array, array_copy));
268
+ EXPECT_EQ (array.Data (), array_copy.Data ());
269
269
270
270
const Array2<T> &array_copy_const = array;
271
271
const Array2<T> &array_copy_const2 = array_copy_const.Generic ().template Specialize <T>();
272
- K2_CHECK_EQ (array_copy_const.Data (), array_copy.Data ());
272
+ EXPECT_EQ (array_copy_const.Data (), array_copy.Data ());
273
+ EXPECT_EQ (array_copy_const2.Data (), array_copy.Data ());
273
274
274
275
const Array1<T> &array1_copy_const = arr1;
275
276
const Array1<T> &array1_copy_const2 = array1_copy_const.Generic ().template Specialize <T>();
276
- K2_CHECK_EQ (array1_copy_const.Data (), array1_copy.Data ());
277
+ EXPECT_EQ (array1_copy_const.Data (), arr1.Data ());
278
+ EXPECT_EQ (array1_copy_const2.Data (), arr1.Data ());
277
279
278
280
279
281
auto cpu_array = array.To (cpu);
0 commit comments