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_read_2I(self): hfi = HarFileIO.readHarFileInfo(TestHarFileIO._dd + "test.har") intaheader = HarFileIO.readHeader(hfi, "INTA") self.assertEqual(intaheader["array"][0, 0], 0) self.assertEqual(intaheader["array"][0, 1], 1) self.assertEqual(intaheader["array"][1, 0], 4) self.assertEqual(intaheader["array"][3, 3], 15)
def test_read_2D_RE(self): hfi = HarFileIO.readHarFileInfo(TestHarFileIO._dd + "test.har") nh01header = HarFileIO.readHeader(hfi, "NH01") self.assertTrue(np.isclose(nh01header["array"][0, 0], 1.0)) self.assertTrue(np.isclose(nh01header["array"][0, 1], 3.14)) self.assertTrue(np.isclose(nh01header["array"][1, 0], 3.14)) self.assertTrue(np.isclose(nh01header["array"][1, 1], 5.0))
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")
def test_read_1C_2(self): hfi = HarFileIO.readHarFileInfo(TestHarFileIO._dd + "test_read.har") 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(header["array"], test_array)]))
def test_read_RE(self): hfi = HarFileIO.readHarFileInfo(TestHarFileIO._dd + "test.har") arr7header = HarFileIO.readHeader(hfi, "ARR7") self.assertTrue( np.isclose(arr7header["array"][0, 0, 0, 0, 0, 0, 0], 1.0)) self.assertTrue( np.isclose(arr7header["array"][0, 0, 0, 0, 0, 0, 1], 2.0)) self.assertTrue( np.isclose(arr7header["array"][0, 0, 0, 0, 0, 1, 0], 3.0)) self.assertTrue( np.isclose(arr7header["array"][0, 0, 0, 0, 1, 0, 0], 4.0)) self.assertTrue( np.isclose(arr7header["array"][0, 0, 0, 1, 0, 0, 0], 5.0)) self.assertTrue( np.isclose(arr7header["array"][0, 0, 1, 0, 0, 0, 0], 6.0)) self.assertTrue( np.isclose(arr7header["array"][0, 1, 0, 0, 0, 0, 0], 7.0)) self.assertTrue( np.isclose(arr7header["array"][1, 0, 0, 0, 0, 0, 0], 8.0)) self.assertTrue( np.isclose(arr7header["array"][1, 1, 1, 1, 1, 1, 1], 9.0))
def test_read_1C(self): hfi = HarFileIO.readHarFileInfo(TestHarFileIO._dd + "test.har") header = HarFileIO.readHeader(hfi, "SIMP") self.assertTrue( all([x == y for (x, y) in zip(header["array"], ["A", "B"])]))