Ejemplo n.º 1
0
    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)
Ejemplo n.º 2
0
 def handle(self, *args, **options):
     main(datetime.strptime(options['date'], '%Y-%m-%d'), num_days=options['num_days'])