feat: order price_average field

Author: T. Ismael Verdugo

cryptomarket/dataclasses/order.py

@@ -20,6 +20,7 @@ class Order:
     created_at: str
     updated_at: str
     price: Optional[str] = None
+    price_average: Optional[str] = None
     expire_time: Optional[str] = None
     stop_price: Optional[str] = None
     post_only: Optional[bool] = None

tests/websockets/test_helpers.py

@@ -1,13 +1,21 @@
 import time
 from dataclasses import asdict
-from typing import Any, Dict, List
+from typing import Any, Callable, Dict, List
 
 from cryptomarket.dataclasses import (Balance, OrderBookLevel, Report,
                                       WSCandle, WSMiniTicker, WSOrderBook,
                                       WSOrderBookTop, WSPublicTrade, WSTicker,
                                       WSTrade)
+from cryptomarket.dataclasses.commission import Commission
 from cryptomarket.dataclasses.wsPriceRate import WSPriceRate
-from tests.rest.test_helpers import good_list
+
+
+def good_list(check_fn: Callable[[Any], bool], list: List[Any]) -> bool:
+    for elem in list:
+        if not check_fn(elem):
+            print(elem)
+            return False
+    return True
 
 
 def defined(a_dict, key):
@@ -19,6 +27,28 @@ def defined(a_dict, key):
     return True
 
 
+def good_balance(balance: Balance) -> bool:
+    return good_dict(
+        asdict(balance),
+        [
+            "currency",
+            "available",
+            "reserved",
+        ]
+    )
+
+
+def good_trading_commission(commission: Commission) -> bool:
+    return good_dict(
+        asdict(commission),
+        [
+            "symbol",
+            "take_rate",
+            "make_rate",
+        ]
+    )
+
+
 def good_dict(a_dict: Dict[str, Any], fields: List[str]) -> bool:
     if not isinstance(a_dict, dict):
         return False

tests/websockets/test_market_data_subs.py

@@ -7,8 +7,7 @@
                                       WSOrderBookTop, WSTicker, WSTrade)
 from cryptomarket.dataclasses.wsPriceRate import WSPriceRate
 from cryptomarket.websockets import MarketDataClient
-from tests.rest.test_helpers import good_list
-from tests.websockets.test_helpers import Veredict, good_candle_list, good_mini_ticker, good_orderbook_top, good_price_rate, good_wsorder_book, good_wsticker, good_wstrade
+from test_helpers import *
 
 SECOND = 1
 MINUTE = 60

tests/websockets/test_trading_client.py

@@ -7,7 +7,6 @@
 from test_helpers import *
 
 from cryptomarket.websockets import TradingClient
-from tests.rest.test_helpers import good_balance, good_trading_commission
 
 with open('/home/ismael/cryptomarket/keys.json') as fd:
     keys = json.load(fd)