class SideTestSuite(unittest.TestCase):
"""Side test cases."""
def setUp(self):
# Object creation
self.tick = Tick(OrderId(TraderId('0'), OrderNumber(1)),
AssetPair(AssetAmount(60, 'BTC'), AssetAmount(30, 'MB')),
Timeout(100), Timestamp.now(), True)
self.tick2 = Tick(OrderId(TraderId('1'), OrderNumber(2)),
AssetPair(AssetAmount(120, 'BTC'), AssetAmount(30, 'MB')),
Timeout(100), Timestamp.now(), True)
self.side = Side()
def test_max_price(self):
# Test max price (list)
self.assertEquals(None, self.side.get_max_price('MB', 'BTC'))
self.assertEquals(None, self.side.get_max_price_list('MB', 'BTC'))
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.assertEquals(Price(0.5, 'MB', 'BTC'), self.side.get_max_price('MB', 'BTC'))
def test_min_price(self):
# Test min price (list)
self.assertEquals(None, self.side.get_min_price_list('MB', 'BTC'))
self.assertEquals(None, self.side.get_min_price('MB', 'BTC'))
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.assertEquals(Price(0.25, 'MB', 'BTC'), self.side.get_min_price('MB', 'BTC'))
def test_insert_tick(self):
# Test insert tick
self.assertEquals(0, len(self.side))
self.assertFalse(self.side.tick_exists(OrderId(TraderId('0'), OrderNumber(1))))
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.assertEquals(2, len(self.side))
self.assertTrue(self.side.tick_exists(OrderId(TraderId('0'), OrderNumber(1))))
def test_remove_tick(self):
# Test remove tick
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.side.remove_tick(OrderId(TraderId('0'), OrderNumber(1)))
self.assertEquals(1, len(self.side))
self.side.remove_tick(OrderId(TraderId('1'), OrderNumber(2)))
self.assertEquals(0, len(self.side))
def test_get_price_level_list_wallets(self):
"""
Test the price level lists of wallets of a side
"""
self.assertFalse(self.side.get_price_level_list_wallets())
self.side.insert_tick(self.tick)
self.assertTrue(self.side.get_price_level_list_wallets())
def test_get_list_representation(self):
"""
Testing the list representation of a side
"""
self.assertFalse(self.side.get_list_representation())
self.side.insert_tick(self.tick)
list_rep = self.side.get_list_representation()
self.assertTrue(list_rep)
class SideTestSuite(unittest.TestCase):
"""Side test cases."""
def setUp(self):
# Object creation
self.tick = Tick(OrderId(TraderId('0'), OrderNumber(1)),
Price(400, 'BTC'), Quantity(30, 'MC'),
Timeout(float("inf")), Timestamp(float("inf")), True)
self.tick2 = Tick(OrderId(TraderId('1'), OrderNumber(2)),
Price(800, 'BTC'), Quantity(30, 'MC'),
Timeout(float("inf")), Timestamp(float("inf")), True)
self.side = Side()
def test_max_price(self):
# Test max price (list)
self.assertEquals(None, self.side.get_max_price('BTC', 'MC'))
self.assertEquals(None, self.side.get_max_price_list('BTC', 'MC'))
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.assertEquals('30.000000 MC\t@\t800.000000 BTC\n',
str(self.side.get_max_price_list('BTC', 'MC')))
self.assertEquals(Price(800, 'BTC'),
self.side.get_max_price('BTC', 'MC'))
def test_min_price(self):
# Test min price (list)
self.assertEquals(None, self.side.get_min_price_list('BTC', 'MC'))
self.assertEquals(None, self.side.get_min_price('BTC', 'MC'))
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.assertEquals('30.000000 MC\t@\t400.000000 BTC\n',
str(self.side.get_min_price_list('BTC', 'MC')))
self.assertEquals(Price(400, 'BTC'),
self.side.get_min_price('BTC', 'MC'))
def test_insert_tick(self):
# Test insert tick
self.assertEquals(0, len(self.side))
self.assertFalse(
self.side.tick_exists(OrderId(TraderId('0'), OrderNumber(1))))
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.assertEquals(2, len(self.side))
self.assertTrue(
self.side.tick_exists(OrderId(TraderId('0'), OrderNumber(1))))
def test_remove_tick(self):
# Test remove tick
self.side.insert_tick(self.tick)
self.side.insert_tick(self.tick2)
self.side.remove_tick(OrderId(TraderId('0'), OrderNumber(1)))
self.assertEquals(1, len(self.side))
self.side.remove_tick(OrderId(TraderId('1'), OrderNumber(2)))
self.assertEquals(0, len(self.side))
def test_get_price_level_list_wallets(self):
"""
Test the price level lists of wallets of a side
"""
self.assertFalse(self.side.get_price_level_list_wallets())
self.side.insert_tick(self.tick)
self.assertTrue(self.side.get_price_level_list_wallets())
def test_get_list_representation(self):
"""
Testing the list representation of a side
"""
self.assertFalse(self.side.get_list_representation())
self.side.insert_tick(self.tick)
list_rep = self.side.get_list_representation()
self.assertTrue(list_rep)
