def test_setup_ex_assembly(self): assem = exo.assembly(name='Unit_test', type='EX_ASSEMBLY', id=444) assem.entity_list = [100, 222] ctype_assem = exo.setup_ex_assembly(assem) self.assertIsInstance(ctype_assem, exo.ex_assembly) self.assertIsInstance(ctype_assem.entity_list, ctypes.POINTER(ctypes.c_longlong))
def test_get_assembly(self): temp_exofile = exo.exodus(self.temp_exo_path) assembly_ids = temp_exofile.get_ids("EX_ASSEMBLY") assemblies = [ temp_exofile.get_assembly(assembly) for assembly in assembly_ids ] root = exo.assembly(name='Root', type='EX_ASSEMBLY', id=100) root.entity_list = [100, 200, 300, 400] self.assertEqual(str(root), str(assemblies[0]))
def test_put_assemblies_multiple_assemblies(self): new = exo.assembly(name='Unit_test1', type='EX_ASSEMBLY', id=444) new.entity_list = [100, 222] new2 = exo.assembly(name='Unit_test2', type='EX_ASSEMBLY', id=448) new2.entity_list = [102, 224] self.temp_exofile.put_assemblies([new, new2]) self.temp_exofile.close() temp_exofile = exo.exodus(self.temp_exo_path) assembly_ids = temp_exofile.get_ids("EX_ASSEMBLY") assemblies = [ temp_exofile.get_assembly(assembly) for assembly in assembly_ids ] self.assertEqual(new.name, assemblies[6].name) self.assertEqual(16, assemblies[6].type) self.assertEqual(new.id, assemblies[6].id) self.assertEqual(new.entity_list, assemblies[6].entity_list) self.assertEqual(new2.name, assemblies[7].name) self.assertEqual(16, assemblies[7].type) self.assertEqual(new2.id, assemblies[7].id) self.assertEqual(new2.entity_list, assemblies[7].entity_list)
def test_add_assembly(self): new = exo.assembly(name='Unit_test', type='EX_ASSEMBLY', id=444) new.entity_list = [100, 222] self.temp_exofile.put_assembly(new) self.temp_exofile.close() temp_exofile = exo.exodus(self.temp_exo_path) assembly_ids = temp_exofile.get_ids("EX_ASSEMBLY") assemblies = [ temp_exofile.get_assembly(assembly) for assembly in assembly_ids ] self.assertEqual(new.name, assemblies[6].name) self.assertEqual('assembly', assemblies[6].type) self.assertEqual(new.id, assemblies[6].id) self.assertEqual(new.entity_list, assemblies[6].entity_list)
def test_put_assemblies(self): new = exo.assembly(name='Unit_test', type='EX_ASSEMBLY', id=444) new.entity_list = [100, 222] with exo.exodus(self.temp_exo_path, mode='a') as temp_exofile: temp_exofile.put_assemblies([new]) with exo.exodus(self.temp_exo_path) as temp_exofile: assembly_ids = temp_exofile.get_ids("EX_ASSEMBLY") assemblies = [ temp_exofile.get_assembly(assembly) for assembly in assembly_ids ] self.assertEqual(new.name, assemblies[6].name) self.assertEqual(16, assemblies[6].type) self.assertEqual(new.id, assemblies[6].id) self.assertEqual(new.entity_list, assemblies[6].entity_list)
def test_copy_opened_in_append_mode(self): new = exo.assembly(name='Unit_test', type=exo.ex_entity_type.EX_ASSEMBLY, id=444) new.entity_list = [100, 222] temp_exo_path2 = self.temp_exo_path + '2' with exo.exodus(self.temp_exo_path, mode='r') as temp_exofile: expected = [assem for assem in temp_exofile.get_ids('EX_ASSEMBLY')] with temp_exofile.copy(temp_exo_path2, True, mode='a') as temp_exofile2: temp_exofile2.put_assembly(new) copied_output = [ assem for assem in temp_exofile2.get_ids('EX_ASSEMBLY') ] self.assertNotEqual(temp_exofile.modeChar, temp_exofile2.modeChar) self.assertNotEquals(expected, copied_output)
def test_get_assemblies(self): temp_exofile = exo.exodus(self.temp_exo_path) assembly_ids = temp_exofile.get_ids("EX_ASSEMBLY") assemblies = temp_exofile.get_assemblies(assembly_ids) expected = [ exo.assembly(name='Root', type='EX_ASSEMBLY', id=100), exo.assembly(name='Child2', type='EX_ELEM_BLOCK', id=200), exo.assembly(name='Child3', type='EX_ELEM_BLOCK', id=300), exo.assembly(name='Child4', type='EX_ELEM_BLOCK', id=400), exo.assembly(name='NewAssembly', type='EX_ASSEMBLY', id=222), exo.assembly(name='FromPython', type='EX_ASSEMBLY', id=333) ] for i, x in enumerate(expected): entity_lists = [[100, 200, 300, 400], [10, 11, 12, 13], [14, 15, 16], [10, 16], [100, 200, 300, 400], [100, 222]] x.entity_list = entity_lists[i] self.maxDiff = None self.assertEqual(str(expected), str(assemblies))