1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
/***************************************************************************
* Copyright (c) Johan Mabille, Sylvain Corlay and Wolf Vollprecht *
* Copyright (c) QuantStack *
* *
* Distributed under the terms of the BSD 3-Clause License. *
* *
* The full license is in the file LICENSE, distributed with this software. *
****************************************************************************/
#include "xtensor/xarray.hpp"
#include "xtensor/xbroadcast.hpp"
#include "xtensor/xfixed.hpp"
#include "xtensor/xshape.hpp"
#include "xtensor/xstrides.hpp"
#include "xtensor/xtensor.hpp"
#include "test_common_macros.hpp"
namespace xt
{
TEST(xshape, initializer_dimension)
{
size_t d0 = initializer_dimension<double>::value;
size_t d1 = initializer_dimension<std::initializer_list<double>>::value;
size_t d2 = initializer_dimension<std::initializer_list<std::initializer_list<double>>>::value;
EXPECT_EQ(size_t(0), d0);
EXPECT_EQ(size_t(1), d1);
EXPECT_EQ(size_t(2), d2);
}
TEST(xshape, shape)
{
auto s0 = shape<std::vector<size_t>>(3);
auto s1 = shape<std::vector<size_t>>(std::initializer_list<size_t>{1, 2});
auto s2 = shape<std::vector<size_t>>(
std::initializer_list<std::initializer_list<size_t>>{{1, 2, 4}, {1, 3, 5}}
);
std::vector<size_t> e0 = {};
std::vector<size_t> e1 = {2};
std::vector<size_t> e2 = {2, 3};
EXPECT_EQ(e0, s0);
EXPECT_EQ(e1, s1);
EXPECT_EQ(e2, s2);
}
TEST(xshape, promote_shape)
{
bool expect_v = std::is_same<
dynamic_shape<size_t>,
promote_shape_t<dynamic_shape<size_t>, std::array<size_t, 3>, std::array<size_t, 2>>>::value;
bool expect_a = std::is_same<
std::array<size_t, 3>,
promote_shape_t<std::array<size_t, 2>, std::array<size_t, 3>, std::array<size_t, 2>>>::value;
ASSERT_TRUE(expect_v);
ASSERT_TRUE(expect_a);
}
TEST(xshape, has_shape)
{
std::array<size_t, 2> shape = {2, 3};
xt::xtensor<size_t, 2> A = xt::zeros<size_t>(shape);
ASSERT_TRUE(xt::has_shape(A, shape));
ASSERT_TRUE(xt::has_shape(A, {2, 3}));
}
}
|
