improve vertexai safety setting test case

luxzoli · luxzoli · commit 75bd2b476013 · 2024-06-13T14:50:05.000Z
diff --git a/test/oai/test_gemini.py b/test/oai/test_gemini.py
@@ -76,8 +76,8 @@ def test_vertexai_safety_setting_conversion(gemini_client):
     ]
 
     def compare_safety_settings(converted_safety_settings, expected_safety_settings):
-        for i, converted_setting in enumerate(converted_safety_settings):
-            expected_setting = expected_safety_settings[i]
+        for i, expected_setting in enumerate(expected_safety_settings):
+            converted_setting = converted_safety_settings[i]
             yield expected_setting.to_dict() == converted_setting.to_dict()
 
     settings_comparison = compare_safety_settings(converted_safety_settings, expected_safety_settings)

Original file line number	Diff line number	Diff line change
`@@ -76,8 +76,8 @@ def test_vertexai_safety_setting_conversion(gemini_client):`
`76`	`76`	`]`
`77`	`77`
`78`	`78`	`def compare_safety_settings(converted_safety_settings, expected_safety_settings):`
`79`		`- for i, converted_setting in enumerate(converted_safety_settings):`
`80`		`- expected_setting = expected_safety_settings[i]`
	`79`	`+ for i, expected_setting in enumerate(expected_safety_settings):`
	`80`	`+ converted_setting = converted_safety_settings[i]`
`81`	`81`	`yield expected_setting.to_dict() == converted_setting.to_dict()`
`82`	`82`
`83`	`83`	`settings_comparison = compare_safety_settings(converted_safety_settings, expected_safety_settings)`