def kw_test(self, data_type, data, fmt): name1 = "file1.txt" name2 = "file2.txt" kw = EclKW("TEST", len(data), data_type) i = 0 for d in data: kw[i] = d i += 1 file1 = open(name1, "w") kw.fprintf_data(file1, fmt) file1.close() file2 = open(name2, "w") for d in data: file2.write(fmt % d) file2.close() self.assertFilesAreEqual(name1, name2) self.assertEqual(kw.data_type, data_type)
def test_fprintf_data(self): with TestAreaContext("kw_no_header"): kw = EclKW("REGIONS", 10, EclDataType.ECL_INT) for i in range(len(kw)): kw[i] = i fileH = open("test", "w") kw.fprintf_data(fileH) fileH.close() fileH = open("test", "r") data = [] for line in fileH.readlines(): tmp = line.split() for elm in tmp: data.append(int(elm)) for (v1,v2) in zip(data,kw): self.assertEqual(v1,v2)