def test_close_file(self): if (os.path.exists("foo")): os.remove("foo") f = FileIO("foo", "w") with self.assertRaises(Exception): f.close()
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)
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)