Add test for Trade

1 files changed, 15 insertions, 1 deletions

diff --git a/test.py b/test.py
index daf5fe4..b747617 100644
--- a/test.py
+++ b/test.py
@@ -489,7 +489,21 @@ class TradeTest(unittest.TestCase):
 
     @unittest.skip("TODO")
     def test_values_assertion(self):
-        pass
+        value_from = Amount("BTC", "1.0")
+        value_from.linked_to = Amount("ETH", "10.0")
+        value_to = Amount("BTC", "1.0")
+        trade = portfolioTrade(value_from, value_to, "ETH")
+        self.assertEqual("BTC", trade.base_currency)
+        self.assertEqual("ETH", trade.currency)
+
+        with self.assertRaises(AssertionError):
+            portfolio.Trade(value_from, value_to, "ETC")
+        with self.assertRaises(AssertionError):
+            value_from.linked_to = None
+            portfolio.Trade(value_from, value_to, "ETH")
+        with self.assertRaises(AssertionError):
+            value_from.currency = "ETH"
+            portfolio.Trade(value_from, value_to, "ETH")
 
     @unittest.skip("TODO")
     def test_fetch_fees(self):