Update build script #805

neuml · Nov 1, 2024 · 9739126 · 9739126
1 parent 077ea41
commit 9739126
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 11 deletions.
diff --git a/Makefile b/Makefile
@@ -33,9 +33,7 @@ test:
 # Run tests while calculating code coverage
 coverage:
 	coverage run -m unittest discover -v -k testann -s ${TEST_DIR}
-	coverage run -m unittest discover -v -k testapi.testauthorization -k testapi.testcluster -s ${TEST_DIR}
-	coverage run -m unittest discover -v -k testapi.testencoding -k testapi.testextension -s ${TEST_DIR}
-	coverage run -m unittest discover -v -k testapi.testpipelines -s ${TEST_DIR}
+	coverage run -m unittest discover -v -k testapi -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testapp -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testarchive -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testcloud -s ${TEST_DIR}
@@ -50,7 +48,7 @@ coverage:
 	coverage run -m unittest discover -v -k testpipeline.testimage -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testpipeline.testllm -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testpipeline.testtext -s ${TEST_DIR}
-	coverage run -m unittest discover -v -k testpipeline.testtrainer -s ${TEST_DIR}
+	coverage run -m unittest discover -v -k testpipeline.testtrain -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testscoring -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testserialize -s ${TEST_DIR}
 	coverage run -m unittest discover -v -k testvectors -s ${TEST_DIR}

diff --git a/test/python/testapi/testembeddings.py → test/python/testapi/testapiembeddings.py b/test/python/testapi/testembeddings.py → test/python/testapi/testapiembeddings.py
@@ -59,13 +59,13 @@
     path: sentence-transformers/nli-mpnet-base-v2
     content: True
     functions:
-        - testapi.testembeddings.Elements
+        - testapi.testapiembeddings.Elements
         - name: length
           argcount: 1
-          function: testapi.testembeddings.length
+          function: testapi.testapiembeddings.length
         - name: ann
           function: ann
-    transform: testapi.testembeddings.transform
+    transform: testapi.testapiembeddings.transform
 """
 
 # Configuration for RAG

diff --git a/test/python/testapi/testpipelines.py → test/python/testapi/testapipipeline.py b/test/python/testapi/testpipelines.py → test/python/testapi/testapipipeline.py
@@ -68,7 +68,7 @@
 
 
 # pylint: disable=R0904
-class TestPipelines(unittest.TestCase):
+class TestPipeline(unittest.TestCase):
     """
     API tests for pipelines.
     """
@@ -98,7 +98,7 @@ def setUpClass(cls):
         Create API client on creation of class.
         """
 
-        cls.client = TestPipelines.start()
+        cls.client = TestPipeline.start()
 
         cls.data = [
             "US tops 5 million confirmed virus cases",

diff --git a/test/python/testapi/testworkflow.py → test/python/testapi/testapiworkflow.py b/test/python/testapi/testworkflow.py → test/python/testapi/testapiworkflow.py
@@ -44,8 +44,8 @@
             - action:
                 - labels
                 - nop
-              initialize: testapi.testworkflow.TestInitFinal
-              finalize: testapi.testworkflow.TestInitFinal
+              initialize: testapi.testapiworkflow.TestInitFinal
+              finalize: testapi.testapiworkflow.TestInitFinal
               merge: concat
               args:
                 - [[positive, negative], false, True]