def test_loadFile_wrong_file(self): f1 = FileIO("tests/test1.csv", "r") loaded1 = f1.loadFile() f1.close() f2 = FileIO("tests/test2.csv", "r") loaded2 = f2.loadFile() f2.close() self.assertEqual((loaded1 == loaded2), False)
class Model: def __init__(self, filename=None): self.__sheet = Sheet() self.__cursor = Cursor() self.__io = FileIO(filename) self.__programIsRunning = True self.__lastCommand = "" self.load() def getCursor(self): return self.__cursor def getSheet(self): return self.__sheet def load(self): self.__sheet.loadPublicObject(self.__io.loadFile()) def save(self): self.__io.saveFile(self.__sheet.getPublicObject()) def quit(self): self.__programIsRunning = False def checkProgramExecution(self): return self.__programIsRunning def echoCommand(self, c): self.__lastCommand = c def getLastCommand(self): return self.__lastCommand
def test_loadFile_good_file(self): f = FileIO("tests/test1.csv", "r") loaded = f.loadFile() f.close() reference = [ Cluster( '0', [Point([1, 1, 1]), Point([2, 2, 2]), Point([3.0, 3.0, 3.0])]), Cluster('1', [ Point([1, 1, 1]), Point([1, 1, 1]), Point([1, 1, 1]), Point([1, 1, 1]), Point([1, 1, 1]), Point([1, 1, 1]), ]), Cluster('2', [Point([1, 1, 1])]) ] self.assertEqual(loaded, reference)