Merge pull request #98 from Portkey-AI/chore/formatting

chore: test case formatting
Portkey-AI · Mar 21, 2024 · 6de4f12 · 6de4f12
2 parents 8b1d3df + 6c67b2f
commit 6de4f12
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/tests/test_async_images.py b/tests/test_async_images.py
@@ -163,7 +163,11 @@ async def test_method_loadbalance_with_two_apikeys(
         )
 
         image = await portkey.images.generate(
-            model="dall-e-3", prompt="A cute baby sea otter", n=1, size="1024x1024", response_format="b64_json"
+            model="dall-e-3",
+            prompt="A cute baby sea otter",
+            n=1,
+            size="1024x1024",
+            response_format="b64_json",
         )
 
         assert type(image.data[0].b64_json) is str
@@ -187,7 +191,11 @@ async def test_method_loadbalance_and_fallback(
         )
 
         image = await portkey.images.generate(
-            model="dall-e-3", prompt="A cute baby sea otter", n=1, size="1024x1024", response_format="b64_json"
+            model="dall-e-3",
+            prompt="A cute baby sea otter",
+            n=1,
+            size="1024x1024",
+            response_format="b64_json",
         )
 
         assert type(image.data[0].b64_json) is str

diff --git a/tests/test_images.py b/tests/test_images.py
@@ -157,7 +157,10 @@ def test_method_loadbalance_with_two_apikeys(
         )
 
         image = portkey.images.generate(
-            prompt="A cute baby sea otter", n=1, size="1024x1024", response_format="b64_json"
+            prompt="A cute baby sea otter",
+            n=1,
+            size="1024x1024",
+            response_format="b64_json",
         )
 
         assert type(image.data[0].b64_json) is str
@@ -178,7 +181,10 @@ def test_method_loadbalance_and_fallback(self, client: Any, config: Dict) -> Non
         )
 
         image = portkey.images.generate(
-            prompt="A cute baby sea otter", n=1, size="1024x1024", response_format="b64_json"
+            prompt="A cute baby sea otter",
+            n=1,
+            size="1024x1024",
+            response_format="b64_json",
         )
 
         assert type(image.data[0].b64_json) is str