Skip to content

Commit

Permalink
Remove unnecessary remote test
Browse files Browse the repository at this point in the history
  • Loading branch information
woodthom2 committed May 31, 2024
1 parent f8995c6 commit 8af3aea
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/remote_tests/test_different_frameworks_remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,6 +178,8 @@ def test_google_multilingual(self):

self.assertLess(0.92, response.json()["matches"][0][2])

# Disabled the vanilla OpenAI as we do not have credentials for this on the live server.
'''
def test_openai(self):
payload_openai = json.loads(json.dumps(json_data_to_match_gad_7))
payload_openai["parameters"]["framework"] = "openai"
Expand All @@ -186,6 +188,7 @@ def test_openai(self):
response = requests.post(endpoint, headers=headers, json=payload_openai)
self.assertLess(0.99, response.json()["matches"][0][0])
'''

def test_azure_openai(self):
payload_openai = json.loads(json.dumps(json_data_to_match_gad_7))
Expand All @@ -196,6 +199,7 @@ def test_azure_openai(self):

self.assertLess(0.99, response.json()["matches"][0][0])

'''
def test_azure_openai_same_as_vanilla_openai(self):
payload_vanilla_openai = json.loads(json.dumps(json_data_to_match_gad_7))
payload_vanilla_openai["parameters"]["framework"] = "openai"
Expand All @@ -218,7 +222,7 @@ def test_azure_openai_same_as_vanilla_openai(self):
self.assertGreater(0.01,
abs(vanilla_openai_response.json()["matches"][0][2] -
azure_openai_response.json()["matches"][0][2]))

'''

if __name__ == '__main__':
unittest.main()

0 comments on commit 8af3aea

Please sign in to comment.