fix: index and search tests

tattle-made · Feb 21, 2024 · ffc8efc · ffc8efc
1 parent 91c797b
commit ffc8efc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/src/tests/test_index_api_as_client.py b/src/tests/test_index_api_as_client.py
@@ -50,15 +50,15 @@ def testIndexImage(self):
             "metadata": {"domain": "hate_speech", "type": ["gender", "caste"]},
             "config": {"mode": "store", "version": "0.1"},
         }
-        with open("sample_data/simple-text.txt", "rb") as media_file:
-            files = {
-                "media": media_file,
-                "data": json.dumps(data),
-            }
-            response = requests.post(url, json=data, headers=headers)
-            # response = requests.post(url, json=data, files=files, headers=headers)
-            print(response.text)
-            self.assertEqual(response.status_code, 200)
+        # with open("sample_data/simple-text.txt", "rb") as media_file:
+        #     files = {
+        #         "media": media_file,
+        #         "data": json.dumps(data),
+        #     }
+        response = requests.post(url, json=data, headers=headers)
+        # response = requests.post(url, json=data, files=files, headers=headers)
+        print(response.text)
+        self.assertEqual(response.status_code, 200)
 
     # @skip
     def testIndexVideo(self):

diff --git a/src/tests/test_search_api_as_client.py b/src/tests/test_search_api_as_client.py
@@ -28,7 +28,7 @@ def testSearchRawQuery(self):
     def testSearchImage(self):
         url = API_URL + "/search"
         data = {"query_type": "image"}
-        with open("sample_data/people.jpg", "rb") as file:
+        with open("tests/sample_data/people.jpg", "rb") as file:
             data = {"data": json.dumps(data)}
             files = {"media": file}
             response = requests.post(url, data=data, files=files)
@@ -39,7 +39,7 @@ def testSearchImage(self):
     # @skip
     def testIndexVideo(self):
         url = API_URL + "/search"
-        with open("sample_data/sample-cat-video.mp4", "rb") as file:
+        with open("tests/sample_data/sample-cat-video.mp4", "rb") as file:
             data = {"data": json.dumps({"query_type": "video"})}
             files = {"media": file}
             response = requests.post(url, data=data, files=files)