fixed overlapping test cases

landing-ai · Mar 16, 2024 · e4935c2 · e4935c2
1 parent 7bfb5c1
commit e4935c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_lmm.py b/tests/test_lmm.py
@@ -51,7 +51,7 @@ def test_generate_classifier(openai_lmm_mock): # noqa: F811
  ['{"Parameters": {"prompt": "cat"}}'],
  indirect=["openai_lmm_mock"],
 )
-def test_generate_classifier(openai_lmm_mock): # noqa: F811
+def test_generate_detector(openai_lmm_mock): # noqa: F811
  lmm = OpenAILMM()
  prompt = "Can you generate a cat classifier?"
  detector = lmm.generate_detector(prompt)
@@ -64,7 +64,7 @@ def test_generate_classifier(openai_lmm_mock): # noqa: F811
  ['{"Parameters": {"prompt": "cat"}}'],
  indirect=["openai_lmm_mock"],
 )
-def test_generate_classifier(openai_lmm_mock): # noqa: F811
+def test_generate_segmentor(openai_lmm_mock): # noqa: F811
  lmm = OpenAILMM()
  prompt = "Can you generate a cat classifier?"
  segmentor = lmm.generate_segmentor(prompt)