How to use test_function_state method in localstack

Best Python code snippet using localstack_python

policy_test.py

Source:policy_test.py

...95        print(a)96        self.assertTrue(boxspace.contains(a))97        self.assertArrayShape(a, (3, 5))98        self.assertArraySubdtypeFloat(a)99    def test_function_state(self):100        env = Env(boxspace, discrete)101        pi = Policy(func_discrete, env, random_seed=19)102        print(pi.function_state)103        batch_norm_avg = pi.function_state['batch_norm/~/mean_ema']['average']104        self.assertArrayShape(batch_norm_avg, (1, 8))105        self.assertArrayNotEqual(batch_norm_avg, jnp.zeros_like(batch_norm_avg))106    def test_bad_input_signature(self):107        def badfunc(S, is_training, x):108            pass109        msg = (110            r"func has bad signature; "111            r"expected: func\(S, is_training\), got: func\(S, is_training, x\)"112        )113        with self.assertRaisesRegex(TypeError, msg):...

stochastic_v_test.py

Source:stochastic_v_test.py

...65        v = StochasticV(func, env, value_range, num_bins=num_bins, random_seed=19)66        v_ = v.mode(s)67        print(v_, env.observation_space)68        self.assertIn(v_, Box(*value_range, shape=()))69    def test_function_state(self):70        env = Env(discrete, discrete)71        value_range = (-10, 10)72        v = StochasticV(func, env, value_range, num_bins=num_bins, random_seed=19)73        print(v.function_state)74        batch_norm_avg = v.function_state['batch_norm/~/mean_ema']['average']75        self.assertArrayShape(batch_norm_avg, (1, 8))76        self.assertArrayNotEqual(batch_norm_avg, jnp.zeros_like(batch_norm_avg))77    # other tests ##################################################################################78    def test_bad_input_signature(self):79        def badfunc(S, is_training, x):80            pass81        msg = (82            r"func has bad signature; "83            r"expected: func\(S, is_training\), "...

v_test.py

Source:v_test.py

...41        v_targ.params = jax.tree_map(jnp.zeros_like, v.params)42        expected = jax.tree_map(lambda a: jnp.full_like(a, tau), v.params)43        v_targ.soft_update(v, tau=tau)44        self.assertPytreeAlmostEqual(v_targ.params, expected)45    def test_function_state(self):46        print(self.v.function_state)47        batch_norm_avg = self.v.function_state['batch_norm/~/mean_ema']['average']48        self.assertArrayShape(batch_norm_avg, (1, 8))49        self.assertArrayNotEqual(batch_norm_avg, jnp.zeros_like(batch_norm_avg))50    def test_bad_input_signature(self):51        def badfunc(S, is_training, x):52            pass53        msg = (54            r"func has bad signature; "55            r"expected: func\(S, is_training\), got: func\(S, is_training, x\)")56        with self.assertRaisesRegex(TypeError, msg):57            V(badfunc, self.env_discrete, random_seed=13)58    def test_bad_output_type(self):59        def badfunc(S, is_training):...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.