portfolio make base done

quatrope · Apr 26, 2023 · a910b28 · a910b28
1 parent f857425
commit a910b28
Showing 1 changed file with 80 additions and 48 deletions.
diff --git a/tests/datasets/test_base.py b/tests/datasets/test_base.py
@@ -9,83 +9,115 @@
 # IMPORTS
 # =============================================================================
 
-# from garpar import datasets
+from garpar import datasets
 
-# import numpy as np
+import numpy as np
 
-# import pytest
+import pytest
 
 
 # =============================================================================
 # TESTS
 # =============================================================================
 
 
-# def test_PorfolioMakerABC_get_window_loss_probability_not_implemethed():
-#     class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
-#         def get_window_loss_probability(self, window_size, entropy):
-#             return super().get_window_loss_probability(window_size, entropy)
+def test_PortfolioMakerABC_make_portfolio_missing_argument_on_definition():
+    with pytest.raises(TypeError):
 
-#         def make_stock_price(self, price, loss, random):
-#             return price
+        class PortfolioMaker(datasets.PortfolioMakerABC):
+            def make_portfolio(self):
+                pass
 
-#     maker = PortfolioMaker()
 
-#     with pytest.raises(NotImplementedError):
-#         maker.make_portfolio()
+def test_PortfolioMakerABC_make_portfolio_missing_not_imp():
+    class PortfolioMaker(datasets.PortfolioMakerABC):
+        def make_portfolio(
+            self,
+            *,
+            window_size=5,
+            days=365,
+            stocks=10,
+            price=100,
+            weights=None,
+        ):
+            return super().make_portfolio(
+                window_size=window_size,
+                days=days,
+                stocks=stocks,
+                price=price,
+                weights=weights,
+            )
 
+    maker = PortfolioMaker()
+    with pytest.raises(NotImplementedError):
+        maker.make_portfolio()
 
-# def test_PorfolioMakerABC_make_stock_price_not_implemethed():
-#     class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
-#         def get_window_loss_probability(self, window_size, entropy):
-#             return 0.2
 
-#         def make_stock_price(self, price, loss, random):
-#             return super().make_stock_price(price, loss, random)
+def test_RandomEntropyPorfolioMakerABC_get_window_loss_probability_not_imp():
+    class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
+        def get_window_loss_probability(self, window_size, entropy):
+            return super().get_window_loss_probability(window_size, entropy)
 
-#     maker = PortfolioMaker()
+        def make_stock_price(self, price, loss, random):
+            return price
 
-#     with pytest.raises(NotImplementedError):
-#         maker.make_portfolio()
+    maker = PortfolioMaker()
 
+    with pytest.raises(NotImplementedError):
+        maker.make_portfolio()
 
-# def test_PorfolioMakerABC_window_size_le_0():
-#     class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
-#         def get_window_loss_probability(self, window_size, entropy):
-#             return 0.2
 
-#         def make_stock_price(self, price, loss, random):
-#             return price
+def test_RandomEntropyPorfolioMakerABC_make_stock_price_not_imp():
+    class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
+        def get_window_loss_probability(self, window_size, entropy):
+            return 0.2
 
-#     maker = PortfolioMaker()
+        def make_stock_price(self, price, loss, random):
+            return super().make_stock_price(price, loss, random)
 
-#     with pytest.raises(ValueError):
-#         maker.make_portfolio(window_size=0)
+    maker = PortfolioMaker()
 
+    with pytest.raises(NotImplementedError):
+        maker.make_portfolio()
 
-# def test_PorfolioMakerABC_window_days_lt_window_size():
-#     class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
-#         def get_window_loss_probability(self, window_size, entropy):
-#             return 0.2
 
-#         def make_stock_price(self, price, loss, random):
-#             return price
+def test_RandomEntropyPorfolioMakerABC_window_size_le_0():
+    class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
+        def get_window_loss_probability(self, window_size, entropy):
+            return 0.2
 
-#     maker = PortfolioMaker()
+        def make_stock_price(self, price, loss, random):
+            return price
 
-#     with pytest.raises(ValueError):
-#         maker.make_portfolio(window_size=5, days=4)
+    maker = PortfolioMaker()
 
+    with pytest.raises(ValueError):
+        maker.make_portfolio(window_size=0)
 
-# def test_PorfolioMakerABC_invalid_number_of_prices():
-#     class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
-#         def get_window_loss_probability(self, window_size, entropy):
-#             return 0.2
 
-#         def make_stock_price(self, price, loss, random):
-#             return price
+def test_RandomEntropyPorfolioMakerABC_window_days_lt_window_size():
+    class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
+        def get_window_loss_probability(self, window_size, entropy):
+            return 0.2
 
-#     maker = PortfolioMaker()
+        def make_stock_price(self, price, loss, random):
+            return price
 
-#     with pytest.raises(ValueError):
-#         maker.make_portfolio(stocks=2, price=[1])
+    maker = PortfolioMaker()
+
+    with pytest.raises(ValueError):
+        maker.make_portfolio(window_size=5, days=4)
+
+
+def test_RandomEntropyPorfolioMakerABC_invalid_number_of_prices():
+    class PortfolioMaker(datasets.RandomEntropyPortfolioMakerABC):
+        def get_window_loss_probability(self, window_size, entropy):
+            return 0.2
+
+        def make_stock_price(self, price, loss, random):
+            return price
+
+    maker = PortfolioMaker()
+
+    with pytest.raises(ValueError):
+        maker.make_portfolio(stocks=2, price=[1])