Skip to content

Commit 582cda1

Browse files
committed
fix dims in test:
1 parent c059a7e commit 582cda1

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

test/model_read/model_read.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -68,23 +68,23 @@ TEST(ExtractValuesFromJsonTests, HandlesNestedArray) {
6868
TEST(ParseJsonShapeTests, HandlesEmptyArray) {
6969
json j = json::array({});
7070
std::vector<size_t> shape;
71-
parse_json_shape(j, shape);
71+
parse_json_shape(j, shape, 0);
7272
std::vector<size_t> expected = {0};
7373
EXPECT_EQ(shape, expected);
7474
}
7575

7676
TEST(ParseJsonShapeTests, HandlesSimpleArray) {
7777
json j = json::array({{1.0, 2.0, 3.0}, {1.0, 2.0, 3.0}});
7878
std::vector<size_t> shape;
79-
parse_json_shape(j, shape);
79+
parse_json_shape(j, shape, 0);
8080
std::vector<size_t> expected = {3};
8181
EXPECT_EQ(shape, expected);
8282
}
8383

8484
TEST(ParseJsonShapeTests, HandlesNestedArray) {
8585
json j = json::array({{{1.0, 2.0}, {3.0, 4.0}}, {3.0, 4.0}});
8686
std::vector<size_t> shape;
87-
parse_json_shape(j, shape);
87+
parse_json_shape(j, shape, 0);
8888
std::vector<size_t> expected = {2, 2};
8989
EXPECT_EQ(shape, expected);
9090
}
@@ -157,7 +157,7 @@ TEST(CreateTensorFromJsonTest, SimpleTensorCheckNoBias) {
157157
TEST(CreateTensorFromJsonTest, EmptyShape) {
158158
json j = json::array({});
159159
std::vector<size_t> shape;
160-
parse_json_shape(j, shape);
160+
parse_json_shape(j, shape, 0);
161161
std::vector<size_t> expected = {0};
162162
EXPECT_EQ(shape, expected);
163163
}
@@ -171,7 +171,7 @@ TEST(ParseJsonShapeTests, HandlesNonArrayJson) {
171171
json j = "not_an_array";
172172
std::vector<size_t> shape;
173173

174-
parse_json_shape(j, shape);
174+
parse_json_shape(j, shape, 0);
175175

176176
std::vector<size_t> expected = {0};
177177
EXPECT_EQ(shape, expected);

0 commit comments

Comments
 (0)