def test_dictionaryImport(self):
cf = CifFile()
cf.import_mmcif_data_map(self.raw_dictionary)
self.assertIsNotNone(cf.getDataBlock("TEST_BLOCK_2"),
"CifFile failed to import dictionary")
self.assertIsInstance(
cf.getDataBlock("TEST_BLOCK_2"),
DataBlock,
"CifFile.__init__ failed to import dictionary",
)
self.assertEqual(
cf.getDataBlock("TEST_BLOCK_2").getCategory(
"_test_category_1").getItem("test_value_1").value,
[1, 2, 3, 4],
"CifFile failed to import dictionary",
)
def test_write_ciffile_after_ciffile_dictionary_import(self):
# Test write CifFile initialized by dictionary
unit_test_file = "io_testcase_1.cif"
cfw = CifFileWriter()
cif_file = CifFile(os.path.join(self.FILE_ROOT, unit_test_file))
cif_file.import_mmcif_data_map(self.raw_dictionary)
cfw.write(cif_file)
cfr = CifFileReader(input="data", preserve_order=True)
test_file = cfr.read(os.path.join(self.FILE_ROOT, unit_test_file),
output="cif_wrapper")
data_block_ids = list(test_file.keys())
data_block_ids.sort()
self.assertEqual(
data_block_ids,
["TEST_BLOCK_1", "TEST_BLOCK_2"],
"Datablock(s) were not written correctly",
)
self.assertEqual(
test_file["TEST_BLOCK_1"]._test_category_2.test_value_1,
["1", "2", "3", "4"],
"mmCIF data was not written correctly",
)