Renamed some tests

neurallayer · Jun 1, 2024 · f52173b · f52173b
1 parent 34185d9
commit f52173b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/samples/alpaca_feed.py b/samples/alpaca_feed.py
@@ -1,21 +1,20 @@
 # %%
-from datetime import timedelta
 
-from roboquant.feeds import AggregatorFeed, get_sp500_symbols
+from roboquant.feeds import get_sp500_symbols
 from roboquant.alpaca import AlpacaLiveFeed
+from roboquant import Timeframe
 
 # %%
-alpaca_feed = AlpacaLiveFeed()
+feed = AlpacaLiveFeed()
 
 stocks = get_sp500_symbols()[:30]
-alpaca_feed.subscribe_quotes(*stocks)
+feed.subscribe_quotes(*stocks)
 
 # alpaca_feed.subscribe_bars(*stocks)
 # feed.subscribe_trades("BTC/USD", "ETH/USD")
 # feed.subscribe("SPXW240312C05190000")
 
-feed = AggregatorFeed(alpaca_feed, timedelta(seconds=15), price_type="quote")
-
-channel = feed.play_background()
+timeframe = Timeframe.next(minutes=1)
+channel = feed.play_background(timeframe)
 while event := channel.get():
-    print(event.items)
+    print(event.time, event.items)
diff --git a/samples/forwardtest_alpaca2.py → samples/forwardtest_alpaca_bars.py b/samples/forwardtest_alpaca2.py → samples/forwardtest_alpaca_bars.py
diff --git a/samples/forwardtest_alpaca.py → samples/forwardtest_alpaca_trades.py b/samples/forwardtest_alpaca.py → samples/forwardtest_alpaca_trades.py