예제 #1
0
파일: tests.py 프로젝트: mgax/MentionGraphs
 def test_cache_filename(self):
     from crawl import CachingMentionCounter, to_epoch
     cmc = CachingMentionCounter(keyword='python',
                                 resolution=timedelta(hours=1),
                                 cache_root='/tmp/cache/path')
     self.assertEqual(cmc.cache_filename_base(date(2011, 9, 3)),
                      '/tmp/cache/path/python/2011-09-03')
예제 #2
0
파일: tests.py 프로젝트: mgax/MentionGraphs
    def test_caching(self):
        from crawl import CachingMentionCounter, to_epoch
        from tempfile import mkdtemp
        import shutil

        day = date(2011, 9, 3)
        self.api_fixture([_mention(to_epoch(day) + 60 * 20, 'english')])

        cache_root = mkdtemp()
        try:
            cmc = CachingMentionCounter(keyword='python',
                                        resolution=timedelta(days=1),
                                        cache_root=cache_root)
            stats1 = cmc.count(day)
            stats2 = cmc.count(day)

        finally:
            shutil.rmtree(cache_root)

        stats_ok = {datetime.combine(day, time()): {
            ('sentiment', None): 1,
            ('generator', 'twitter'): 1,
            ('language', 'english'): 1,
            None: 1
        }}
        self.assertEqual(stats1, stats_ok)
        self.assertEqual(stats2, stats_ok)
        self.assertEqual(self._mock_call.call_count, 1)