Fix timeframe comparisons

MDUYN · MDUYN · commit f42c35122739 · 2024-07-04T11:01:29.000+02:00
diff --git a/tests/infrastructure/market_data_sources/test_ccxt_ohlcv_market_data_source.py b/tests/infrastructure/market_data_sources/test_ccxt_ohlcv_market_data_source.py
@@ -62,4 +62,4 @@ def test_get_timeframe(self):
             market="BITVAVO",
             symbol="BTC/EUR",
         )
-        self.assertEqual("15m", ccxt_ohlcv_market_data_source.timeframe.value)
+        self.assertEqual("15m", ccxt_ohlcv_market_data_source.timeframe)
diff --git a/tests/infrastructure/market_data_sources/test_csv_ohlcv_market_data_source.py b/tests/infrastructure/market_data_sources/test_csv_ohlcv_market_data_source.py
@@ -196,4 +196,4 @@ def test_get_timeframe(self):
             timeframe="2h",
             window_size=10,
         )
-        self.assertEqual(TimeFrame.TWO_HOUR, datasource.get_timeframe())
+        self.assertTrue(TimeFrame.TWO_HOUR.equals(datasource.get_timeframe()))

Original file line number	Diff line number	Diff line change
`@@ -62,4 +62,4 @@ def test_get_timeframe(self):`
`62`	`62`	`market="BITVAVO",`
`63`	`63`	`symbol="BTC/EUR",`
`64`	`64`	`)`
`65`		`- self.assertEqual("15m", ccxt_ohlcv_market_data_source.timeframe.value)`
	`65`	`+ self.assertEqual("15m", ccxt_ohlcv_market_data_source.timeframe)`
Original file line number	Diff line number	Diff line change
`@@ -196,4 +196,4 @@ def test_get_timeframe(self):`
`196`	`196`	`timeframe="2h",`
`197`	`197`	`window_size=10,`
`198`	`198`	`)`
`199`		`- self.assertEqual(TimeFrame.TWO_HOUR, datasource.get_timeframe())`
	`199`	`+ self.assertTrue(TimeFrame.TWO_HOUR.equals(datasource.get_timeframe()))`