diff --git a/tests/test_cellxgene.py b/tests/test_cellxgene.py index 80efd58a..12019820 100644 --- 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 index d0276596..58bce75a 100644 --- 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."}}],