Skip to content

Commit 0e74776

Browse files
committed
fixing tests
1 parent 087b977 commit 0e74776

1 file changed

Lines changed: 30 additions & 14 deletions

File tree

test/test_xadapt.cpp

Lines changed: 30 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,39 +19,55 @@ namespace xt
1919
TEST(xarray_adaptor, xarray_pointer)
2020
{
2121
std::vector<int> data(4, 0);
22-
using shape_type = xt::svector<std::size_t>;
23-
shape_type shape({2, 2});
24-
22+
xt::svector<std::size_t> shape = {2, 2};
2523
xt::xarray_pointer<int> a = xt::adapt(data.data(), data.size(), xt::no_ownership(), shape);
2624

2725
a(0, 0) = 0;
2826
a(0, 1) = 1;
2927
a(1, 0) = 2;
3028
a(1, 1) = 3;
3129

32-
EXPECT_EQ(0, data[0]);
33-
EXPECT_EQ(1, data[1]);
34-
EXPECT_EQ(2, data[2]);
35-
EXPECT_EQ(3, data[3]);
30+
if (XTENSOR_DEFAULT_LAYOUT == xt::layout_type::row_major)
31+
{
32+
EXPECT_EQ(0, data[0]);
33+
EXPECT_EQ(1, data[1]);
34+
EXPECT_EQ(2, data[2]);
35+
EXPECT_EQ(3, data[3]);
36+
}
37+
else
38+
{
39+
EXPECT_EQ(0, data[0]);
40+
EXPECT_EQ(2, data[1]);
41+
EXPECT_EQ(1, data[2]);
42+
EXPECT_EQ(3, data[3]);
43+
}
3644
}
3745

3846
TEST(xtensor_adaptor, xtensor_pointer)
3947
{
4048
std::vector<int> data(4, 0);
41-
using shape_type = std::array<std::size_t, 2>;
42-
shape_type shape({2, 2});
43-
49+
std::array<std::size_t, 2> shape = {2, 2};
4450
xt::xtensor_pointer<int, 2> a = xt::adapt(data.data(), data.size(), xt::no_ownership(), shape);
4551

4652
a(0, 0) = 0;
4753
a(0, 1) = 1;
4854
a(1, 0) = 2;
4955
a(1, 1) = 3;
5056

51-
EXPECT_EQ(0, data[0]);
52-
EXPECT_EQ(1, data[1]);
53-
EXPECT_EQ(2, data[2]);
54-
EXPECT_EQ(3, data[3]);
57+
if (XTENSOR_DEFAULT_LAYOUT == xt::layout_type::row_major)
58+
{
59+
EXPECT_EQ(0, data[0]);
60+
EXPECT_EQ(1, data[1]);
61+
EXPECT_EQ(2, data[2]);
62+
EXPECT_EQ(3, data[3]);
63+
}
64+
else
65+
{
66+
EXPECT_EQ(0, data[0]);
67+
EXPECT_EQ(2, data[1]);
68+
EXPECT_EQ(1, data[2]);
69+
EXPECT_EQ(3, data[3]);
70+
}
5571
}
5672

5773
TEST(xarray_adaptor, adapt)

0 commit comments

Comments
 (0)