class TestTradeChain(unittest.TestCase):
def setUp(self):
self.chain = TradeChain()
def test_profit_functions(self):
some_market = "A market name"
self.chain.add_trade(Trade(some_market, "ask", 32.8
).trade_from(3.28, "USD"
).trade_to(0.1, "BTC")
)
self.chain.add_trade(Trade(some_market, "ask", 0.05
).trade_from(0.1, "BTC"
).trade_to(2, "LTC")
)
self.chain.add_trade(Trade(some_market, "bid", 2
).trade_from(2, "LTC"
).trade_to(4, "USD")
)
assert float(self.chain.profit) == 0.72
assert float(self.chain.percentage) == 21.95
def setUp(self):
self.chain = TradeChain()
