#53 Fix test failing

Author: pockerman

src/tests/test_sarsa_semi_gradient.py

@@ -1,7 +1,7 @@
 import unittest
 import pytest
 from src.algorithms.sarsa_semi_gradient import SARSAnConfig, SARSAn
-from src.spaces.tiled_environment import TiledEnv
+from src.spaces.tiled_environment import TiledEnv, TiledEnvConfig
 from src.policies.epsilon_greedy_policy import EpsilonGreedyPolicy, EpsilonDecayOption
 from src.exceptions.exceptions import InvalidParamValue
 
@@ -18,15 +18,29 @@ def test_actions_before_training_throws_invalid_environment(self):
                              "satisfy the IS_TILED_ENV_CONSTRAINT constraint", str(e))
 
     def test_actions_before_training_throws_invalid_policy(self):
-
-            env = TiledEnv(env=None, tiling_dim=10, num_tilings=4096, max_size=100)
-            config = SARSAnConfig()
-            agent = SARSAn(sarsa_config=config)
-            with pytest.raises(InvalidParamValue) as e:
-                agent.actions_before_training(env=env)
+        env_config = TiledEnvConfig()
+        env_config.max_size = 4096
+        env_config.num_tilings = 5
+        env_config.tiling_dim = 6
+        env_config.env = None
+        env_config.column_scales = {"col1": [0.0, 1.0]}
+
+        env = TiledEnv(env_config)
+        config = SARSAnConfig()
+        agent = SARSAn(sarsa_config=config)
+        with pytest.raises(InvalidParamValue) as e:
+            agent.actions_before_training(env=env)
 
     def test_actions_before_training_throws_estimator_not_set(self):
-        env = TiledEnv(env=None, tiling_dim=10, num_tilings=4096, max_size=100)
+
+        env_config = TiledEnvConfig()
+        env_config.max_size = 4096
+        env_config.num_tilings = 5
+        env_config.tiling_dim = 6
+        env_config.env = None
+        env_config.column_scales = {"col1": [0.0, 1.0]}
+
+        env = TiledEnv(env_config)
         policy = EpsilonGreedyPolicy(eps=1.0, n_actions=1, decay_op=EpsilonDecayOption.NONE)
         config = SARSAnConfig()
         config.policy = policy