def test_simple_run(self): agg_instance = ModelAggregator(TestModel) agg_instance.run() self.assertEqual(kaggregate.get_aggregate_value("count"), 10) self.assertEqual(kaggregate.get_aggregate_value("sum"), 45) self.assertEqual(kaggregate.get_aggregate_value("average"), 4.5)
def test_simple_run_with_prefix(self): agg_instance = TestAggregator() agg_instance.run("foo_") self.assertEqual(kaggregate.get_aggregate_value("foo_count"), 10) self.assertEqual(kaggregate.get_aggregate_value("foo_sum"), 45) self.assertEqual(kaggregate.get_aggregate_value("average"), None) self.assertEqual(kaggregate.get_aggregate_value("count"), None)
def test_management_command(self): management.call_command('gen_aggregates') self.assertEqual(kaggregate.get_aggregate_value("count"), 10) self.assertEqual(kaggregate.get_aggregate_value("sum"), 45) self.assertEqual(kaggregate.get_aggregate_value("bar_count"), 10) self.assertEqual(kaggregate.get_aggregate_value("bar_sum"), 45) self.assertEqual(kaggregate.get_aggregate_value("foo-count"), 10) self.assertEqual(kaggregate.get_aggregate_value("foo-sum"), 45) self.assertEqual(kaggregate.get_aggregate_value("foo-average"), 4.5)
def test_simple_run(self): agg_instance = TestAggregator() agg_instance.run() self.assertEqual(kaggregate.get_aggregate_value("count"), 10) self.assertEqual(kaggregate.get_aggregate_value("sum"), 45)