@@ -68,23 +68,23 @@ TEST(ExtractValuesFromJsonTests, HandlesNestedArray) {
68
68
TEST (ParseJsonShapeTests, HandlesEmptyArray) {
69
69
json j = json::array ({});
70
70
std::vector<size_t > shape;
71
- parse_json_shape (j, shape);
71
+ parse_json_shape (j, shape, 0 );
72
72
std::vector<size_t > expected = {0 };
73
73
EXPECT_EQ (shape, expected);
74
74
}
75
75
76
76
TEST (ParseJsonShapeTests, HandlesSimpleArray) {
77
77
json j = json::array ({{1.0 , 2.0 , 3.0 }, {1.0 , 2.0 , 3.0 }});
78
78
std::vector<size_t > shape;
79
- parse_json_shape (j, shape);
79
+ parse_json_shape (j, shape, 0 );
80
80
std::vector<size_t > expected = {3 };
81
81
EXPECT_EQ (shape, expected);
82
82
}
83
83
84
84
TEST (ParseJsonShapeTests, HandlesNestedArray) {
85
85
json j = json::array ({{{1.0 , 2.0 }, {3.0 , 4.0 }}, {3.0 , 4.0 }});
86
86
std::vector<size_t > shape;
87
- parse_json_shape (j, shape);
87
+ parse_json_shape (j, shape, 0 );
88
88
std::vector<size_t > expected = {2 , 2 };
89
89
EXPECT_EQ (shape, expected);
90
90
}
@@ -157,7 +157,7 @@ TEST(CreateTensorFromJsonTest, SimpleTensorCheckNoBias) {
157
157
TEST (CreateTensorFromJsonTest, EmptyShape) {
158
158
json j = json::array ({});
159
159
std::vector<size_t > shape;
160
- parse_json_shape (j, shape);
160
+ parse_json_shape (j, shape, 0 );
161
161
std::vector<size_t > expected = {0 };
162
162
EXPECT_EQ (shape, expected);
163
163
}
@@ -171,7 +171,7 @@ TEST(ParseJsonShapeTests, HandlesNonArrayJson) {
171
171
json j = " not_an_array" ;
172
172
std::vector<size_t > shape;
173
173
174
- parse_json_shape (j, shape);
174
+ parse_json_shape (j, shape, 0 );
175
175
176
176
std::vector<size_t > expected = {0 };
177
177
EXPECT_EQ (shape, expected);
0 commit comments