Trying to fix gpt and cellxgene installation for testing

pachterlab · May 15, 2023 · 279bed2 · 279bed2
1 parent 00bd401
commit 279bed2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/tests/test_cellxgene.py b/tests/test_cellxgene.py
@@ -37,11 +37,8 @@ def repr_dict(adata):
 
 
 class TestCellxgene(unittest.TestCase):
-    def install_dependencies(self):
-        module = "cellxgene"
-        gget_setup(module)
-
     def test_cellxgene_adata(self):
+        gget_setup("cellxgene")
         test = "test1"
         expected_result = cellxgene_dict[test]["expected_result"]
         result_to_test = cellxgene(**cellxgene_dict[test]["args"])
@@ -52,6 +49,7 @@ def test_cellxgene_adata(self):
         self.assertEqual(result_to_test, expected_result)
 
     def test_cellxgene_metadata(self):
+        gget_setup("cellxgene")
         test = "test2"
         expected_result = cellxgene_dict[test]["expected_result"]
         result_to_test = cellxgene(**cellxgene_dict[test]["args"])

diff --git a/tests/test_gpt.py b/tests/test_gpt.py
@@ -5,12 +5,10 @@
 
 
 class TestGpt(unittest.TestCase):
-    def install_dependencies(self):
-        module = "gpt"
-        gget_setup(module)
-
     @patch("openai.ChatCompletion.create")
     def test_gpt(self, mock_create):
+        gget_setup("gpt")
+
         # Mock the response from the API
         mock_response = {
             "choices": [{"message": {"content": "This is a generated response."}}],