Fix timezone in tests

diff --git a/tests/helper.py b/tests/helper.py
index a2f8a2243c5e6966a96ccda794741291970d201a..930d04a4cc17f86fbe945794a2716834f07f332e 100644 (file)
--- a/tests/helper.py
+++ b/tests/helper.py
@@ -5,9 +5,10 @@ from unittest import mock
 import requests_mock
 from io import StringIO
 import portfolio, market, main, store, dbs
 import requests_mock
 from io import StringIO
 import portfolio, market, main, store, dbs

+import datetime

 
 __all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",
 
 __all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",

-        "StringIO"]
+        "StringIO", "tz"]

 
 limits = ["acceptance", "unit"]
 for test_type in limits:
 
 limits = ["acceptance", "unit"]
 for test_type in limits:


@@ -19,6 +20,9 @@ for test_type in limits:
         limits = [test_type]
         break
 
         limits = [test_type]
         break
 

+def tz(hours):
+    return datetime.timezone(datetime.timedelta(hours=hours))
+

 class WebMockTestCase(unittest.TestCase):
     import time
 
 class WebMockTestCase(unittest.TestCase):
     import time
 

diff --git a/tests/test_portfolio.py b/tests/test_portfolio.py
index cad3095a34eb2bfd0f42ef1b398d6ad878b86a6c..f2f639470c0bbf880916d16f5d0583b59392357e 100644 (file)
--- a/tests/test_portfolio.py
+++ b/tests/test_portfolio.py
@@ -1361,7 +1361,7 @@ class OrderTest(WebMockTestCase):
         with self.subTest(similar_open_order=True):
             order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
                     D("0.1"), "BTC", "long", self.m, "trade")
         with self.subTest(similar_open_order=True):
             order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
                     D("0.1"), "BTC", "long", self.m, "trade")

-            order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55)
+            order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55, 0, tz(2))

 
             self.m.ccxt.order_precision.return_value = 8
             self.m.ccxt.fetch_orders.return_value = [
 
             self.m.ccxt.order_precision.return_value = 8
             self.m.ccxt.fetch_orders.return_value = [


@@ -1471,7 +1471,7 @@ class OrderTest(WebMockTestCase):
         with self.subTest(similar_open_order=False, past_trades=True):
             order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
                     D("0.1"), "BTC", "long", self.m, "trade")
         with self.subTest(similar_open_order=False, past_trades=True):
             order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
                     D("0.1"), "BTC", "long", self.m, "trade")

-            order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55)
+            order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55, 0, tz(2))

 
             self.m.ccxt.order_precision.return_value = 8
             self.m.ccxt.fetch_orders.return_value = []
 
             self.m.ccxt.order_precision.return_value = 8
             self.m.ccxt.fetch_orders.return_value = []