def test_main(self): _dt = datetime(2014, 01, 01) patches = { 'topic_extraction' : DEFAULT, 'topic_clustering' : DEFAULT } expected = { 'end_date': '2014-01-02', 'n_clusters': 6, 'top_n': 3, 'cluster_topics': [], 'date': '2014-01-01', 'start_date': '2013-12-18', 'num_days': 15 } with patch.multiple('topic_analysis.main', **patches) as mocks: ret = main(_dt) self.assertEquals(expected, ret)
def handle(self, *args, **options): main(datetime.strptime(options['date'], '%Y-%m-%d'), num_days=options['num_days'])