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