aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests/helper.py6
-rw-r--r--tests/test_portfolio.py4
2 files changed, 7 insertions, 3 deletions
diff --git a/tests/helper.py b/tests/helper.py
index a2f8a22..930d04a 100644
--- a/tests/helper.py
+++ b/tests/helper.py
@@ -5,9 +5,10 @@ from unittest import mock
5import requests_mock 5import requests_mock
6from io import StringIO 6from io import StringIO
7import portfolio, market, main, store, dbs 7import portfolio, market, main, store, dbs
8import datetime
8 9
9__all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D", 10__all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",
10 "StringIO"] 11 "StringIO", "tz"]
11 12
12limits = ["acceptance", "unit"] 13limits = ["acceptance", "unit"]
13for test_type in limits: 14for test_type in limits:
@@ -19,6 +20,9 @@ for test_type in limits:
19 limits = [test_type] 20 limits = [test_type]
20 break 21 break
21 22
23def tz(hours):
24 return datetime.timezone(datetime.timedelta(hours=hours))
25
22class WebMockTestCase(unittest.TestCase): 26class WebMockTestCase(unittest.TestCase):
23 import time 27 import time
24 28
diff --git a/tests/test_portfolio.py b/tests/test_portfolio.py
index cad3095..f2f6394 100644
--- a/tests/test_portfolio.py
+++ b/tests/test_portfolio.py
@@ -1361,7 +1361,7 @@ class OrderTest(WebMockTestCase):
1361 with self.subTest(similar_open_order=True): 1361 with self.subTest(similar_open_order=True):
1362 order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"), 1362 order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
1363 D("0.1"), "BTC", "long", self.m, "trade") 1363 D("0.1"), "BTC", "long", self.m, "trade")
1364 order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55) 1364 order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55, 0, tz(2))
1365 1365
1366 self.m.ccxt.order_precision.return_value = 8 1366 self.m.ccxt.order_precision.return_value = 8
1367 self.m.ccxt.fetch_orders.return_value = [ 1367 self.m.ccxt.fetch_orders.return_value = [
@@ -1471,7 +1471,7 @@ class OrderTest(WebMockTestCase):
1471 with self.subTest(similar_open_order=False, past_trades=True): 1471 with self.subTest(similar_open_order=False, past_trades=True):
1472 order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"), 1472 order = portfolio.Order("buy", portfolio.Amount("ETH", "0.001"),
1473 D("0.1"), "BTC", "long", self.m, "trade") 1473 D("0.1"), "BTC", "long", self.m, "trade")
1474 order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55) 1474 order.start_date = datetime.datetime(2018, 3, 25, 15, 15, 55, 0, tz(2))
1475 1475
1476 self.m.ccxt.order_precision.return_value = 8 1476 self.m.ccxt.order_precision.return_value = 8
1477 self.m.ccxt.fetch_orders.return_value = [] 1477 self.m.ccxt.fetch_orders.return_value = []