class TestPassengers(TestCase): def setUp(self): self.pas = Passengers('test_passengers_data.csv') def test_sum_male(self): self.assertEquals(self.pas.sum_male(), 1) def test_sum_female(self): self.assertEquals(self.pas.sum_female(), 4) def test_percent_of_surviving(self): self.assertEquals(self.pas.percent_of_surviving(), 50) def test_mean_age_of_passenger(self): self.assertEquals(self.pas.mean_age_of_passenger(), 3) def test_median_age_of_passenger(self): self.assertEquals(self.pas.median_age_of_passenger(), 2) def test_percent_of_first_class(self): self.assertEquals(self.pas.percent_of_first_class(), 50) def test_most_popular_female_name(self): self.assertEquals(self.pas.most_popular_female_name(), ('Mary', 3)) def test_probability_of_surviving(self): self.assertEquals(self.pas.probability_of_surviving(2, 1, 'female'), 47.222222222222214)