Skip to content

Commit

Permalink
[ML] Fix main() functions for test suites (#2383)
Browse files Browse the repository at this point in the history
These have had a silly mistake ever since they were written.
The observer was unregistered after main() returned. Clearly
it didn't matter, as the program was about to exit anyway,
but it's nice to tidy up the code.
  • Loading branch information
droberts195 authored Aug 3, 2022
1 parent c1eefad commit 5f57234
Show file tree
Hide file tree
Showing 10 changed files with 30 additions and 10 deletions.
4 changes: 3 additions & 1 deletion bin/controller/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion bin/pytorch_inference/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/api/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/core/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/maths/analytics/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/maths/common/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/maths/time_series/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/model/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/seccomp/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}
4 changes: 3 additions & 1 deletion lib/ver/unittest/Main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
int main(int argc, char** argv) {
ml::test::CTestObserver observer;
boost::unit_test::framework::register_observer(observer);
return boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init, argc, argv);
int result{boost::unit_test::unit_test_main(&ml::test::CBoostTestXmlOutput::init,
argc, argv)};
boost::unit_test::framework::deregister_observer(observer);
return result;
}

0 comments on commit 5f57234

Please sign in to comment.