Unverified Commit f29aba8c authored by Binyao Jiang's avatar Binyao Jiang Committed by GitHub
Browse files
parent faa25df1
...@@ -96,8 +96,13 @@ class TestOpenAIVisionServer(CustomTestCase): ...@@ -96,8 +96,13 @@ class TestOpenAIVisionServer(CustomTestCase):
), f"text: {text}, should contain cab, taxi, SUV, vehicle or car" ), f"text: {text}, should contain cab, taxi, SUV, vehicle or car"
# MiniCPMO fails to recognize `iron`, but `hanging` # MiniCPMO fails to recognize `iron`, but `hanging`
assert ( assert (
"iron" in text or "hang" in text or "cloth" in text or "holding" in text "iron" in text
), f"text: {text}, should contain iron, hang, cloth or holding" or "hang" in text
or "cloth" in text
or "coat" in text
or "holding" in text
or "outfit" in text
), f"text: {text}, should contain iron, hang, cloth, coat or holding or outfit"
assert response.id assert response.id
assert response.created assert response.created
assert response.usage.prompt_tokens > 0 assert response.usage.prompt_tokens > 0
...@@ -193,11 +198,15 @@ class TestOpenAIVisionServer(CustomTestCase): ...@@ -193,11 +198,15 @@ class TestOpenAIVisionServer(CustomTestCase):
print(f"Multi images response:\n{text}") print(f"Multi images response:\n{text}")
print("-" * 30) print("-" * 30)
assert ( assert (
"man" in text or "cab" in text or "SUV" in text or "taxi" in text "man" in text
), f"text: {text}, should contain man, cab, SUV or taxi" or "cab" in text
or "SUV" in text
or "taxi" in text
or "car" in text
), f"text: {text}, should contain man, cab, SUV, taxi or car"
assert ( assert (
"logo" in text or '"S"' in text or "SG" in text "logo" in text or '"S"' in text or "SG" in text or "graphic" in text
), f"text: {text}, should contain logo, S or SG" ), f"text: {text}, should contain logo, S or SG or graphic"
assert response.id assert response.id
assert response.created assert response.created
assert response.usage.prompt_tokens > 0 assert response.usage.prompt_tokens > 0
...@@ -320,11 +329,12 @@ class TestOpenAIVisionServer(CustomTestCase): ...@@ -320,11 +329,12 @@ class TestOpenAIVisionServer(CustomTestCase):
or "individual" in video_response or "individual" in video_response
or "speaker" in video_response or "speaker" in video_response
or "Steve" in video_response or "Steve" in video_response
or "hand" in video_response
), f""" ), f"""
====================== video_response ===================== ====================== video_response =====================
{video_response} {video_response}
=========================================================== ===========================================================
should contain 'man' or 'person' or 'individual' or 'speaker' should contain 'man' or 'person' or 'individual' or 'speaker' or 'hand'
""" """
assert ( assert (
"present" in video_response "present" in video_response
...@@ -375,7 +385,8 @@ class TestOpenAIVisionServer(CustomTestCase): ...@@ -375,7 +385,8 @@ class TestOpenAIVisionServer(CustomTestCase):
or "person" in video_response or "person" in video_response
or "individual" in video_response or "individual" in video_response
or "speaker" in video_response or "speaker" in video_response
), f"video_response: {video_response}, should either have 'man' in video_response, or 'person' in video_response, or 'individual' in video_response or 'speaker' in video_response" or "hand" in video_response
), f"video_response: {video_response}, should either have 'man' in video_response, or 'person' in video_response, or 'individual' in video_response, or 'speaker' in video_response or 'hand' in video_response"
assert ( assert (
"present" in video_response "present" in video_response
or "examine" in video_response or "examine" in video_response
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment