def test_get_random_stimulus(self): """ Does environment gives proper random stimuli - form given sequence """ stimuli = range(10) env = Environment(stimuli) for _ in range(100): self.assertTrue(env.get_stimulus() in stimuli)