remove data provider implementation / cleanup

notadamking · Jul 2, 2019 · 6bf7dbf · 6bf7dbf
1 parent a795969
commit 6bf7dbf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 9 deletions.
diff --git a/lib/data/providers/ExchangeDataProvider.py b/lib/data/providers/ExchangeDataProvider.py
@@ -1,8 +1,7 @@
-import time
 import ccxt
 import pandas as pd
 
-from typing import Dict, Tuple
+from typing import Tuple
 from datetime import datetime
 
 from lib.data.providers.dates import ProviderDateFormat

diff --git a/test/data/test_providers.py b/test/data/test_providers.py
@@ -1,7 +1,7 @@
 import pytest
 
 from lib.data.providers.dates import ProviderDateFormat
-from lib.data.providers import StaticDataProvider, ExchangeDataProvider
+from lib.data.providers import StaticDataProvider
 
 
 @pytest.fixture
@@ -28,7 +28,7 @@ def test_column_map(self, csv_provider):
     def test_date_sort(self, csv_provider):
         ohlcv = csv_provider.historical_ohlcv()
 
-        timestamps = ohlcv['Timestamp'].values
+        timestamps = ohlcv['Date'].values
         sorted_timestamps = sorted(timestamps.copy())
 
         assert (timestamps == sorted_timestamps).all()
diff --git a/test/test_rl_trader.py b/test/test_rl_trader.py
@@ -1,9 +1,7 @@
 import unittest
-from unittest.mock import Mock
 
 from lib.RLTrader import RLTrader
 from lib.TraderArgs import TraderArgs
-from lib.data_feed.StaticDataProvider import StaticDataProvider
 
 
 class TestRLTrader(unittest.TestCase):
@@ -12,11 +10,9 @@ def setUp(self):
 
     def test_that_args_get_injected_correctly(self):
         args = self.parser.parse_args(['optimize'])
-        data_mock = Mock(StaticDataProvider)
-        sut = RLTrader(data_mock, args)
+        sut = RLTrader(args)
 
         self.assertEqual(sut.tensorboard_path, args.tensorboard_path)
         self.assertEqual(sut.params_db_path, args.params_db_path)
         self.assertEqual(sut.model_verbose, args.model_verbose)
         self.assertEqual(sut.nminibatches, args.nminibatches)
-        self.assertEqual(sut.test_set_percentage, args.test_set_percentage)