def test_get_sex_ratio(self): stats = Stats(data_json='{"sex_distribution": {"male": 2000, "female": 1000}}') self.assertEqual(stats.get_sex_ratio(), 2) stats = Stats(data_json='{"sex_distribution": {"female": 1000}}') self.assertEqual(stats.get_sex_ratio(), 0.1) stats = Stats(data_json='{"sex_distribution": {"male": 2000}}') self.assertEqual(stats.get_sex_ratio(), 10) stats = Stats(data_json='{"sex_distribution": {}}') self.assertEqual(stats.get_sex_ratio(), 1) stats = Stats(data_json='[]') self.assertEqual(stats.get_sex_ratio(), 1)
class TestStats(unittest.TestCase): def setUp(self): self.stats = Stats(data_json='{"foo": "bar"}') def test_get_data(self): self.assertEqual(self.stats.get_data(), {'foo': 'bar'}) def test_set_data(self): self.stats.set_data({'bar': 'boo'}) self.assertEqual(self.stats.data_json, '{"bar": "boo"}') def test_get_sex_ratio(self): stats = Stats(data_json='{"sex_distribution": {"male": 2000, "female": 1000}}') self.assertEqual(stats.get_sex_ratio(), 2) stats = Stats(data_json='{"sex_distribution": {"female": 1000}}') self.assertEqual(stats.get_sex_ratio(), 0.1) stats = Stats(data_json='{"sex_distribution": {"male": 2000}}') self.assertEqual(stats.get_sex_ratio(), 10) stats = Stats(data_json='{"sex_distribution": {}}') self.assertEqual(stats.get_sex_ratio(), 1) stats = Stats(data_json='[]') self.assertEqual(stats.get_sex_ratio(), 1)
def setUp(self): self.stats = Stats(data_json='{"foo": "bar"}')
def test_init__some_stats_in_db_2(self): stats1 = Stats.create(data_json='', created=datetime.datetime(1990, 1, 1)) stats2 = Stats.create(data_json='', created=datetime.datetime(1980, 1, 1)) stats_service = StatsService() self.assertEqual(stats_service._stats, stats1)