class TestRuleSource(unittest.TestCase): """ Tests for `RuleSource`. """ def setUp(self): self.source = RuleSource("/") def test_is_data_file_when_is(self): self.assertTrue(self.source.is_data_file("/my/file.rule.py")) def test_is_data_file_when_is_not(self): self.assertFalse(self.source.is_data_file("/my/file.py"))