Skip to content

Commit 87dc377

Browse files
committed
Revert changes in tests
1 parent b1f3e52 commit 87dc377

1 file changed

Lines changed: 7 additions & 9 deletions

File tree

metatomic-torch/tests/models.cpp

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -132,15 +132,13 @@ TEST_CASE("Models metadata") {
132132
auto options = torch::make_intrusive<ModelEvaluationOptionsHolder>();
133133
options->set_length_unit("nanometer");
134134

135-
auto output1 = torch::make_intrusive<ModelOutputHolder>();
136-
output1->per_atom = false;
137-
options->outputs.insert("output_1", output1);
135+
options->outputs.insert("output_1", torch::make_intrusive<ModelOutputHolder>());
138136

139-
auto output2 = torch::make_intrusive<ModelOutputHolder>();
140-
output2->per_atom = true;
141-
output2->set_quantity("something");
142-
output2->set_unit("something");
143-
options->outputs.insert("output_2", output2);
137+
auto output = torch::make_intrusive<ModelOutputHolder>();
138+
output->per_atom = true;
139+
output->set_quantity("something");
140+
output->set_unit("something");
141+
options->outputs.insert("output_2", output);
144142

145143
const auto* expected = R"({
146144
"class": "ModelEvaluationOptions",
@@ -194,7 +192,7 @@ TEST_CASE("Models metadata") {
194192
);
195193
CHECK(*options->get_selected_atoms().value() == *expected_selection);
196194

197-
auto output = options->outputs.at("foo");
195+
output = options->outputs.at("foo");
198196
CHECK(output->quantity().empty());
199197
CHECK(output->unit().empty());
200198
CHECK(output->per_atom == false);

0 commit comments

Comments
 (0)