Fixing unit test.

Team-Kujira · Oct 26, 2023 · b935385 · b935385
1 parent bb74caf
commit b935385
Showing 1 changed file with 6 additions and 12 deletions.
diff --git a/...hummingbot/connector/gateway/clob_spot/data_sources/kujira/test_kujira_api_data_source.py b/...hummingbot/connector/gateway/clob_spot/data_sources/kujira/test_kujira_api_data_source.py
@@ -708,15 +708,9 @@ def test_order_sides(self):
 
     def test_order_types(self):
         for item in KujiraOrderType:
-            if item != KujiraOrderType.MARKET:
-                hummingbot_type = KujiraOrderType.to_hummingbot(item)
-                kujira_type = KujiraOrderType.from_hummingbot(hummingbot_type)
-                kujira_type_from_name = KujiraOrderType.from_name(kujira_type.name)
-
-                self.assertEqual(item, kujira_type)
-                self.assertEqual(item, kujira_type_from_name)
-            else:
-                with self.assertRaises(ValueError) as context:
-                    KujiraOrderType.to_hummingbot(item)
-
-                self.assertEqual(str(context.exception), 'Unrecognized order type "OrderType.MARKET".')
+            hummingbot_type = KujiraOrderType.to_hummingbot(item)
+            kujira_type = KujiraOrderType.from_hummingbot(hummingbot_type)
+            kujira_type_from_name = KujiraOrderType.from_name(kujira_type.name)
+
+            self.assertEqual(item, kujira_type)
+            self.assertEqual(item, kujira_type_from_name)