diff --git a/pcbdevice/tests/utils/test_fileUtils.py b/pcbdevice/tests/utils/test_fileUtils.py index 134536f..81f7d10 100644 --- a/pcbdevice/tests/utils/test_fileUtils.py +++ b/pcbdevice/tests/utils/test_fileUtils.py @@ -6,13 +6,13 @@ from pcbdevice.utils.FileUtils import FileUtils resources = './pcbdevice/tests/resources/' class TestFileUtils(TestCase): - def test_pbmToCsv(self): - actual = FileUtils.pbmToMatrix(resources + 'raw/test1.pbm') + def test_pbmToMatrix(self): + actual, h, w = FileUtils.pbmToMatrix(resources + 'raw/test1.pbm') expected = TestUtils.readIntFile(resources + 'formatted/test1.csv') assert actual == expected def test_saveMatrixToFile(self): - actual = FileUtils.pbmToMatrix(resources + 'raw/test1.pbm') + actual, h, w = FileUtils.pbmToMatrix(resources + 'raw/test1.pbm') FileUtils.saveMatrixToFile(actual, resources + 'output/test1.csv') expected = TestUtils.readIntFile(resources + 'output/test1.csv') assert actual == expected