ADD: Added setter properties to BidAskPair.

Author: anthonyb8

python/python/mbinary/_lib.pyi

@@ -203,6 +203,18 @@ class BidAskPair:
     def bid_ct(self) -> int: ...
     @property
     def ask_ct(self) -> int: ...
+    @bid_px.setter
+    def bid_px(self, value: int) -> None: ...
+    @ask_px.setter
+    def ask_px(self, value: int) -> None: ...
+    @bid_sz.setter
+    def bid_sz(self, value: int) -> None: ...
+    @ask_sz.setter
+    def ask_sz(self, value: int) -> None: ...
+    @ask_ct.setter
+    def ask_ct(self, value: int) -> None: ...
+    @bid_ct.setter
+    def bid_ct(self, value: int) -> None: ...
 
 class OhlcvMsg(RecordMsg):
     def __init__(

python/tests/test_integration.py

@@ -900,6 +900,37 @@ def test_symbol_map(self):
         mappings = symbol_map.map
         self.assertEqual(mappings, mappings)
 
+    def test_bid_ask(self):
+        pair = BidAskPair(1, 2, 3, 4, 5, 6)
+
+        # Validate Initial Data
+        self.assertEqual(pair.bid_px, 1)
+        self.assertEqual(pair.ask_px, 2)
+        self.assertEqual(pair.bid_sz, 3)
+        self.assertEqual(pair.ask_sz, 4)
+        self.assertEqual(pair.bid_ct, 5)
+        self.assertEqual(pair.ask_ct, 6)
+        self.assertEqual(pair.pretty_bid_px, 1 / 1e9)
+        self.assertEqual(pair.pretty_ask_px, 2 / 1e9)
+
+        # Test
+        pair.ask_px = 900
+        pair.bid_px = 800
+        pair.ask_sz = 700
+        pair.bid_sz = 600
+        pair.ask_ct = 500
+        pair.bid_ct = 400
+
+        # Validate 
+        self.assertEqual(pair.bid_px, 800)
+        self.assertEqual(pair.ask_px, 900)
+        self.assertEqual(pair.bid_sz, 600)
+        self.assertEqual(pair.ask_sz, 700)
+        self.assertEqual(pair.bid_ct, 400)
+        self.assertEqual(pair.ask_ct, 500)
+        self.assertEqual(pair.pretty_bid_px, 800 / 1e9)
+        self.assertEqual(pair.pretty_ask_px, 900 / 1e9)
+
     def test_is_record(self):
         pair = BidAskPair(1, 2, 3, 4, 5, 6)
         ohlcv_msg = OhlcvMsg(1, 123456765432, 0, 1, 2, 3, 4, 100000)

rust/src/python/records.rs

@@ -47,6 +47,36 @@ impl BidAskPair {
         }
     }
 
+    #[setter]
+    fn set_bid_px(&mut self, price: i64) {
+        self.bid_px = price;
+    }
+
+    #[setter]
+    fn set_ask_px(&mut self, price: i64) {
+        self.ask_px = price;
+    }
+
+    #[setter]
+    fn set_bid_sz(&mut self, size: u32) {
+        self.bid_sz = size;
+    }
+
+    #[setter]
+    fn set_ask_sz(&mut self, size: u32) {
+        self.ask_sz = size;
+    }
+
+    #[setter]
+    fn set_bid_ct(&mut self, count: u32) {
+        self.bid_ct = count;
+    }
+
+    #[setter]
+    fn set_ask_ct(&mut self, count: u32) {
+        self.ask_ct = count;
+    }
+
     #[getter]
     fn pretty_bid_px(&self) -> f64 {
         self.bid_px as f64 / PRICE_SCALE as f64