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)
