tests: Consolidate test case classes

Author: nitesh585

CHANGELOG.md

@@ -6,6 +6,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 ### Added
+- Consolidate test case classes
 - Tutorial on how to load models dynamically
   https://intel.github.io/dffml/tutorials/models/load.html
 - Added download progressbar in `util/net.py`

tests/df/test_auto_start.py

@@ -14,6 +14,7 @@ async def announce():
 
 class TestAutoStart(AsyncTestCase):
     async def setUp(self):
+        await super().setUp()
         dataflow = DataFlow(
             operations={
                 "announce": announce.op,

tests/operation/test_io.py

@@ -13,7 +13,7 @@
 
 class TestInputOutput(AsyncTestCase):
     async def setUp(self):
-        super().setUp()
+        await super().setUp()
         self.stdout = io.StringIO()
         InputDataflow = DataFlow(
             operations={

tests/operation/test_sqlite_query.py

@@ -29,7 +29,8 @@ def tearDownClass(cls):
         super().tearDownClass()
         os.unlink(cls.database_name)
 
-    def setUp(self):
+    async def setUp(self):
+        await super().setUp()
         self.table_name = "myTable"
         self.cols = {
             "key": "real",

tests/test_df.py

@@ -79,7 +79,8 @@ async def parse_line(line: str):
 
 
 class TestMemoryKeyValueStore(AsyncTestCase):
-    def setUp(self):
+    async def setUp(self):
+        await super().setUp()
         self.kvStore = MemoryKeyValueStore(BaseConfig())
 
     async def test_get_set(self):
@@ -96,12 +97,14 @@ async def test_get_none(self):
 
 class TestMemoryOperationImplementationNetwork(AsyncTestCase):
     async def setUp(self):
+        await super().setUp()
         self.operationsNetwork = MemoryOperationImplementationNetwork.withconfig(
             {}
         )
         self.operationsNetworkCtx = await self.operationsNetwork.__aenter__()
 
     async def tearDown(self):
+        await super().tearDown()
         await self.operationsNetwork.__aexit__(None, None, None)
 
     async def test_contains_true(self):

tests/test_feature.py

@@ -8,7 +8,8 @@
 
 
 class TestFeature(AsyncTestCase):
-    def setUp(self):
+    async def setUp(self):
+        await super().setUp()
         self.feature = Feature("name", int, 1)
 
     def test_default_dtype(self):
@@ -33,7 +34,8 @@ def test_convert_dtype_invalid(self):
 
 
 class TestFeatures(AsyncTestCase):
-    def setUp(self):
+    async def setUp(self):
+        await super().setUp()
         self.one = Feature("one", int, 1)
         self.two = Feature("two", float, 2)
         self.three = Feature("three", int, 1)

tests/test_input_validation.py

@@ -55,6 +55,7 @@ def echo_shout(shout_in):
 
 class TestDefintion(AsyncTestCase):
     async def setUp(self):
+        await super().setUp()
         self.dataflow = DataFlow(
             operations={
                 "get_circle": get_circle.op,