def test_get_new_states_probs_type_error(self): policy = policies.ProductionRuleAppendPolicy(grammar=self.grammar) with self.assertRaisesRegexp( TypeError, r'Input state shoud be an instance of ' r'states\.ProductionRulesState'): policy.get_new_states_probs(states.StateBase())
def setUp(self): super(StateBaseTest, self).setUp() self.state = states.StateBase()