Fix tests

This commit is contained in:
Marc-Antoine Lafreniere
2019-02-16 20:37:17 -05:00
parent ff7a860c2d
commit 2b1feca451

View File

@@ -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