From 7a7a4ec92daa71f18c836a3458e559d5bb3395c8 Mon Sep 17 00:00:00 2001 From: liord Date: Sun, 5 Jan 2025 13:16:41 +0200 Subject: [PATCH] Add test to test_suite --- tests/test_suite.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_suite.py b/tests/test_suite.py index 66a72711c..519c75f8f 100644 --- a/tests/test_suite.py +++ b/tests/test_suite.py @@ -28,7 +28,8 @@ from tests.common_tests.function_tests.test_resource_utilization_object import TestResourceUtilizationObject from tests.common_tests.function_tests.test_threshold_selection import TestThresholdSelection from tests.common_tests.test_doc_examples import TestCommonDocsExamples -from tests.common_tests.test_tp_model import TargetPlatformModelingTest, OpsetTest, QCOptionsTest, FusingTest +from tests.common_tests.test_tp_model import TargetPlatformModelingTest, OpsetTest, QCOptionsTest, FusingTest, \ + TPModelInputOutputTests found_tf = importlib.util.find_spec("tensorflow") is not None if found_tf: @@ -116,6 +117,7 @@ suiteList.append(unittest.TestLoader().loadTestsFromTestCase(TestHistogramCollector)) suiteList.append(unittest.TestLoader().loadTestsFromTestCase(TestCollectorsManipulations)) suiteList.append(unittest.TestLoader().loadTestsFromTestCase(TestThresholdSelection)) + suiteList.append(unittest.TestLoader().loadTestsFromTestCase(TPModelInputOutputTests)) suiteList.append(unittest.TestLoader().loadTestsFromTestCase(TargetPlatformModelingTest)) suiteList.append(unittest.TestLoader().loadTestsFromTestCase(OpsetTest)) suiteList.append(unittest.TestLoader().loadTestsFromTestCase(QCOptionsTest))