Ejemplos de ProcessData.get_csv_names en Python

Ejemplo n.º 1

Archivo: process_data_test.py Proyecto: terpumos/trumpyd

class TestProcessData(TestCase):
    def setUp(self):
        self.process_data = ProcessData('./test/input/',
                                        './test/output/temp_folder/')

    def test___get_csv_names(self):
        names = [
            './test/input/subfolder1/test1.json',
            './test/input/subfolder1/test2.json',
            './test/input/subfolder2/test3.json',
        ]
        test_names = [name for name in self.process_data.get_csv_names()]
        print(test_names)
        self.assertEqual(len(test_names), 3)
        self.assertTrue(all(name in names for name in test_names))

    def tear_down(self):
        pass