Как проверить правильность сохраненного файла csv

У меня есть небольшая функция, которая ничего не возвращает, но вместо этого создает файл csv. Я хотел бы протестировать его, если данный выходной CSV-файл содержит правильные данные. Как я могу этого добиться?

функция

def create_csv(input, output)
    df = pd.read_csv(input, names=['number1', 'number2'])
    df.number1 *= 2
    df.number2 *= 5
    df.to_csv(output, index=False, header=False)

python pandas unit-testing mocking
person maslak    schedule 02.02.2019    source источник
comment
Распечатайте перед сохранением   -  person anky    schedule 02.02.2019
comment
я не хочу это печатать   -  person maslak    schedule 02.02.2019
comment
Что вы передаете параметрам input и output?   -  person wwii    schedule 02.02.2019
comment
пути входного и выходного файла   -  person maslak    schedule 02.02.2019

Ответы (2)


arrow_upward
2
arrow_downward

.to_csv и .read_csv принимают буферы, а также пути строк. Документация по этим методам предлагает использовать io.StringIO в качестве альтернативы открытому файлу или пути к файлу.

В модульном тесте используйте объекты io.StringIO для передачи функции во время теста. Что-то вроде этого для простого теста функции

import unittest, io
class TestOne(unittest.TestCase):
    def setUp(self):
        s = '''3,4\n2,1\n5,6'''
        self.fakecsv = io.StringIO(s)
        self.fakefile = io.StringIO()
        self.result = '''6,20\n4,5\n10,30\n'''

    def test_create(self):
        create_csv(self.fakecsv, self.fakefile)
        self.fakefile.seek(0)
        self.assertEqual(self.fakefile.read(), self.result)
person wwii    schedule 02.02.2019

arrow_upward
2
arrow_downward

Протестируйте фрейм данных непосредственно перед его сохранением в CSV-файле. тестовые сценарии NumPy и тестовые функции Pandas могут оказаться полезными. Если вы считаете, что вам нужно протестировать сам файл CSV, вы на самом деле не доверяете функции to_csv() панд. Тогда лучший способ действий — просмотреть модульные тесты pandas и если вы чувствуете, что этого не хватает, сделайте запрос на извлечение, чтобы внести в него тесты, которые, по вашему мнению, отсутствуют.

person André C. Andersen    schedule 02.02.2019