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)