def test_write_1C(self):
shutil.copy2(TestHarFileIO._dd + "test.har", "temp.har")
hfi = HarFileIO.readHarFileInfo("temp.har")
chst_header = HarFileIO.readHeader(hfi, "CHST")
self.assertTrue(chst_header.is_valid())
chst_header["array"][0] = "F_string" # Change one element
HarFileIO.writeHeaders("temp.har", chst_header)
hfi = HarFileIO.readHarFileInfo("temp.har")
test_hn = ['CHST']
hfi_headers = [ha_info.name for ha_info in hfi["ha_infos"]]
self.assertTrue(all([x == y for (x, y) in zip(hfi_headers, test_hn)]))
chst_header = HarFileIO.readHeader(hfi, "CHST")
test_array = [
"F_string ", "B_string ", "C_string ", "D_string ",
"E_string "
]
self.assertTrue(
all([x == y for (x, y) in zip(chst_header["array"], test_array)]))
os.remove("temp.har")
def test_write_2I(self):
shutil.copy2(TestHarFileIO._dd + "test.har", "temp.har")
hfi = HarFileIO.readHarFileInfo("temp.har")
inta_header = HarFileIO.readHeader(hfi, "INTA")
self.assertTrue(inta_header.is_valid())
inta_header["array"][3, 3] = 30
HarFileIO.writeHeaders("temp.har", inta_header)
hfi = HarFileIO.readHarFileInfo("temp.har")
inta_header = HarFileIO.readHeader(hfi, "INTA")
self.assertEqual(inta_header["array"][3, 3], 30)
os.remove("temp.har")
def test_write_all(self):
# shutil.copy2(TestHarFileIO._dd + "test.har", "temp.har")
hfi = HarFileIO.readHarFileInfo(TestHarFileIO._dd + "test.har")
nh01 = HarFileIO.readHeader(hfi, "NH01")
arr7 = HarFileIO.readHeader(hfi, "ARR7")
HarFileIO.writeHeaders("temp.har", [nh01, arr7])
hfi = HarFileIO.readHarFileInfo("temp.har")
hn = hfi.getHeaderArrayNames()
test_hn = ["NH01", "ARR7"]
self.assertTrue(all([x == y for (x, y) in zip(hn, test_hn)]))
os.remove("temp.har")
def test_write_RE(self):
shutil.copy2(TestHarFileIO._dd + "test.har", "temp.har")
hfi = HarFileIO.readHarFileInfo("temp.har")
arr7_header = HarFileIO.readHeader(hfi, "ARR7")
self.assertTrue(arr7_header.is_valid())
arr7_header["array"][1, 1, 1, 1, 1, 1, 1] = 103.14
HarFileIO.writeHeaders("temp.har", arr7_header)
hfi = HarFileIO.readHarFileInfo("temp.har")
inta_header = HarFileIO.readHeader(hfi, "ARR7")
self.assertTrue(
np.isclose(inta_header["array"][1, 1, 1, 1, 1, 1, 1], 103.14))
os.remove("temp.har")