def test_probability(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(678))
        ant, prob = self.antenna_genetator.get_random_antenna()
        self.assertAlmostEqual(prob, 26**-1)
        self.assertEqual(ant, 2)

        _, prob = self.antenna_genetator.get_random_antenna()
        self.assertAlmostEqual(prob, 21**-1)
class TestSuccesiveAntennaGenerator(TestCase):
    def test_get_random_antenna(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(678))

        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, 2)
        expt = [4,5,6,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24]
        self.assertListEqual(self.antenna_genetator._antennas, expt)

        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, 14)
        expt = [15,16,17,18,20,21,22,23,24]
        self.assertListEqual(self.antenna_genetator._antennas, expt)

        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, -1)
        expt = []
        self.assertListEqual(self.antenna_genetator._antennas, expt)

    def test_get_random_antenna_bottom_line(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(607))
        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, 21)
        expt = [23, 24, ]
        self.assertListEqual(self.antenna_genetator._antennas, expt)

    def test_get_random_antenna_right_line(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(637))
        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, 14)
        expt = [15, 16, 17, 18, 20, 21, 22, 23, 24]
        self.assertListEqual(self.antenna_genetator._antennas, expt)

    def test_get_random_antenna_last_box(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(631))
        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(24, ant)
        expt = []
        self.assertListEqual(self.antenna_genetator._antennas, expt)

    def test_probability(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(678))
        ant, prob = self.antenna_genetator.get_random_antenna()
        self.assertAlmostEqual(prob, 26**-1)
        self.assertEqual(ant, 2)

        _, prob = self.antenna_genetator.get_random_antenna()
        self.assertAlmostEqual(prob, 21**-1)
    def test_get_random_antenna(self):
        self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(678))

        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, 2)
        expt = [4,5,6,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24]
        self.assertListEqual(self.antenna_genetator._antennas, expt)

        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, 14)
        expt = [15,16,17,18,20,21,22,23,24]
        self.assertListEqual(self.antenna_genetator._antennas, expt)

        ant, _ = self.antenna_genetator.get_random_antenna()
        self.assertEqual(ant, -1)
        expt = []
        self.assertListEqual(self.antenna_genetator._antennas, expt)
 def test_get_random_antenna_last_box(self):
     self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(631))
     ant, _ = self.antenna_genetator.get_random_antenna()
     self.assertEqual(24, ant)
     expt = []
     self.assertListEqual(self.antenna_genetator._antennas, expt)
 def test_get_random_antenna_right_line(self):
     self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(637))
     ant, _ = self.antenna_genetator.get_random_antenna()
     self.assertEqual(ant, 14)
     expt = [15, 16, 17, 18, 20, 21, 22, 23, 24]
     self.assertListEqual(self.antenna_genetator._antennas, expt)
 def test_get_random_antenna_bottom_line(self):
     self.antenna_genetator = SuccesiveAntennaGenerator(5, Random(607))
     ant, _ = self.antenna_genetator.get_random_antenna()
     self.assertEqual(ant, 21)
     expt = [23, 24, ]
     self.assertListEqual(self.antenna_genetator._antennas, expt)