⚡ Optimize Bitcoin price simulation with NumPy vectorization

.gitignore

@@ -39,3 +39,7 @@
 
 # debug information files
 *.dwo
+
+# Python
+__pycache__/
+*.pyc

bitcoin_trading_simulation.py

@@ -14,15 +14,20 @@ def simulate_bitcoin_prices(days=60, initial_price=50000, volatility=0.02):
     Simulates Bitcoin prices for a given number of days using Geometric Brownian Motion.
     """
     dt = 1
-    prices = [initial_price]
-    for _ in range(days - 1):
-        # Using a simplified model for daily price changes
-        drift = 0  # Assuming no long-term drift for simplicity
-        shock = np.random.normal(0, volatility)
-        price_change = prices[-1] * (drift * dt + shock * np.sqrt(dt))
-        prices.append(prices[-1] + price_change)
+    # Using a simplified model for daily price changes
+    drift = 0  # Assuming no long-term drift for simplicity
+
+    # Vectorized Geometric Brownian Motion simulation
+    shocks = np.random.normal(0, volatility, days - 1)
+    multipliers = 1 + drift * dt + shocks * np.sqrt(dt)
+
+    prices = np.empty(days)
+    prices[0] = initial_price
+    prices[1:] = initial_price * np.cumprod(multipliers)
+
     return pd.Series(prices, name='Price')
 
+
 def calculate_moving_averages(prices, short_window=7, long_window=30):
     """
     Calculates short and long moving averages for a given price series.

test_bitcoin_trading.py

@@ -0,0 +1,29 @@
+import unittest
+import pandas as pd
+from bitcoin_trading_simulation import simulate_bitcoin_prices
+
+
+class TestBitcoinSimulation(unittest.TestCase):
+    def test_simulate_bitcoin_prices_length(self):
+        days = 100
+        prices = simulate_bitcoin_prices(days=days)
+        self.assertEqual(len(prices), days)
+
+    def test_simulate_bitcoin_prices_start_value(self):
+        initial_price = 50000
+        prices = simulate_bitcoin_prices(initial_price=initial_price)
+        self.assertEqual(prices.iloc[0], initial_price)
+
+    def test_simulate_bitcoin_prices_type(self):
+        prices = simulate_bitcoin_prices()
+        self.assertIsInstance(prices, pd.Series)
+        self.assertEqual(prices.name, 'Price')
+
+    def test_simulate_bitcoin_prices_values(self):
+        # Check that prices are not all the same (volatility implies change)
+        prices = simulate_bitcoin_prices(days=100, volatility=0.02)
+        self.assertGreater(prices.std(), 0)
+
+
+if __name__ == '__main__':
+    unittest.main()