def test_count_one_word(self):
ext = LizardExtension()
list(ext.extend_tokens([("a", 0), ("b", 0)]))
self.assertEqual(1, ext.result['a'])
self.assertEqual(1, ext.result['b'])
def test_should_not_count_keywords(self):
ext = LizardExtension()
list(ext.extend_tokens([("for", 1)]))
self.assertNotIn('for', ext.result)
def test_should_count_non_keyword(self):
ext = LizardExtension()
list(ext.extend_tokens([("For", 1)]))
self.assertIn('For', ext.result)
def test_count_one_word_multiple_times(self):
ext = LizardExtension()
list(ext.extend_tokens([("a", 1), ("a", 1)]))
self.assertEqual(2, ext.result['a'])