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))
