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)