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.getEclType() , data_type )
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.getEclType(), 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)