Esempio n. 1
0
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"))