def test_cache_works_1(self): xs = rules.cached_segment_enriched_tokens(self.s1) rules.cached_segment_enriched_tokens(self.s2) ys = rules.cached_segment_enriched_tokens(self.s1) self.assertTrue(xs is ys)
def test_cache_works_2(self): mockedGET = mock.MagicMock(return_value=self.s1.get_enriched_tokens()) self.s1.get_enriched_tokens = mockedGET rules.cached_segment_enriched_tokens(self.s1) rules.cached_segment_enriched_tokens(self.s1) mockedGET.assert_called_once_with()
def test_simple(self): xs = rules.cached_segment_enriched_tokens(self.s1) self.assertTrue(isinstance(xs, list)) self.assertTrue(all(isinstance(x, RichToken) for x in xs))