示例#1
0
 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)
示例#2
0
    def test_random_stimuli_with_distance(self):
        samples = [Sample([x]) for x in xrange(10)] * 10
        chooser = RandomStimuliChooser(None, True, 3)
        env = Environment(samples, chooser)
        for _ in xrange(10):
            sort = sorted([x.get_values()[0] for x in env.get_stimuli(4)])
            self.assertEqual([0, 3, 6, 9], sort)

        chooser = RandomStimuliChooser(None, True, 5)
        env = Environment(samples, chooser)
        for _ in xrange(10):
            sort = sorted([x.get_values()[0] for x in env.get_stimuli(2)])
            self.assertEqual(len(sort), 2)
            self.assertTrue(sort[0] < 5)
            self.assertTrue(sort[1] >= 5)

        self.assertRaises(Exception, chooser.get_stimuli, samples, 100)
示例#3
0
 def test_get_all_stimuli(self):
     """ Does environment returns proper sequence of stimuli - given earlier
     """
     env = Environment(range(10))
     self.assertEqual(env.get_all_stimuli(),  range(10))