def testCountOnTrack(self): count0 = 0 count1 = 0 for i in range(1, 100): train = TrainTrack(i // 2, i) for j in range(1, 100): list = train.simulate(j)[1:] count0 += list.count(0) count1 += list.count(1) answer = count0 < count1 self.assertEqual(True, answer)
def testLength1(self): train = TrainTrack(1, 1) value = train.simulate(1) self.assertEqual([0], value)
def testZeroL1(self): train = TrainTrack(10, 0) value = 1 in train.simulate(20)[1:] self.assertEqual(True, value)
def testZeroL0(self): train = TrainTrack(0, 10) value = 0 in train.simulate(20)[1:] self.assertEqual(False, value)
def testLength1000(self): train = TrainTrack(10, 10) answer = len(train.simulate(1000)) self.assertEqual(1000, answer)