def test_remove_antenna(self): gen = BadAntennaGenerator(3, Random(666)) gen._remove_antenna(gen._antennas.index(2)) expt = [0,3,4,6,7,8] self.assertListEqual(gen._antennas, expt) gen._remove_antenna(gen._antennas.index(0)) expt = [4,6,7,8] self.assertListEqual(gen._antennas, expt) gen._remove_antenna(gen._antennas.index(7)) expt = [] self.assertListEqual(gen._antennas, expt)
def test_remove_antenna_easy(self): gen = BadAntennaGenerator(3, Random(666)) gen._remove_antenna(2) expt = [0,3,4,6,7,8] self.assertListEqual(gen._antennas, expt) gen = BadAntennaGenerator(3, Random(666)) gen._remove_antenna(3) expt = [1, 2, 5, 7, 8] self.assertListEqual(gen._antennas, expt) gen = BadAntennaGenerator(3, Random(666)) gen._remove_antenna(4) expt = [0,2,6,8] self.assertListEqual(gen._antennas, expt) gen = BadAntennaGenerator(3, Random(666)) gen._remove_antenna(7) expt = [0,1,2,3,5] self.assertListEqual(gen._antennas, expt)