def testWriteDataFile(self):
"""Test case - write data file"""
self.lfh.write(
"\nStarting %s %s\n"
% (self.__class__.__name__, sys._getframe().f_code.co_name)
)
try:
#
myDataList = []
ofh = open(tempname(suffix=".cif"), "w")
curContainer = DataContainer("myblock")
aCat = DataCategory("pdbx_seqtool_mapping_ref")
aCat.appendAttribute("ordinal")
aCat.appendAttribute("entity_id")
aCat.appendAttribute("auth_mon_id")
aCat.appendAttribute("auth_mon_num")
aCat.appendAttribute("pdb_chain_id")
aCat.appendAttribute("ref_mon_id")
aCat.appendAttribute("ref_mon_num")
aCat.append((1, 2, 3, 4, 5, 6, 7))
aCat.append((1, 2, 3, 4, 5, 6, 7))
aCat.append((1, 2, 3, 4, 5, 6, 7))
aCat.append((1, 2, 3, 4, 5, 6, 7))
curContainer.append(aCat)
myDataList.append(curContainer)
pdbxW = PdbxWriter(ofh)
pdbxW.write(myDataList)
ofh.close()
except:
traceback.print_exc(file=sys.stderr)
self.fail()
def testSimpleInitialization(self):
"""Test case - Simple initialization of a data category and data block
"""
self.lfh.write(
"\nStarting %s %s\n" %
(self.__class__.__name__, sys._getframe().f_code.co_name))
try:
#
fn = tempname(suffix='.cif')
attributeNameList = [
'aOne', 'aTwo', 'aThree', 'aFour', 'aFive', 'aSix', 'aSeven',
'aEight', 'aNine', 'aTen'
]
rowList = [[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]]
nameCat = 'myCategory'
#
#
curContainer = DataContainer("myblock")
aCat = DataCategory(nameCat, attributeNameList, rowList)
aCat.printIt()
curContainer.append(aCat)
curContainer.printIt()
#
myContainerList = []
myContainerList.append(curContainer)
ofh = open(fn, "w")
pdbxW = PdbxWriter(ofh)
pdbxW.write(myContainerList)
ofh.close()
myContainerList = []
ifh = open(fn, "r")
pRd = PdbxReader(ifh)
pRd.read(myContainerList)
ifh.close()
for container in myContainerList:
for objName in container.getObjNameList():
name, aList, rList = container.getObj(objName).get()
self.lfh.write("Recovered data category %s\n" % name)
self.lfh.write("Attribute list %r\n" %
repr(aList))
self.lfh.write("Row list %r\n" %
repr(rList))
except:
traceback.print_exc(file=self.lfh)
self.fail()
def testUpdateDataFile(self):
"""Test case - update data file"""
self.lfh.write(
"\nStarting %s %s\n" %
(self.__class__.__name__, sys._getframe().f_code.co_name))
try:
# Create a initial data file --
#
myDataList = []
curContainer = DataContainer("myblock")
aCat = DataCategory("pdbx_seqtool_mapping_ref")
aCat.appendAttribute("ordinal")
aCat.appendAttribute("entity_id")
aCat.appendAttribute("auth_mon_id")
aCat.appendAttribute("auth_mon_num")
aCat.appendAttribute("pdb_chain_id")
aCat.appendAttribute("ref_mon_id")
aCat.appendAttribute("ref_mon_num")
aCat.append([9, 2, 3, 4, 5, 6, 7])
aCat.append([10, 2, 3, 4, 5, 6, 7])
aCat.append([11, 2, 3, 4, 5, 6, 7])
aCat.append([12, 2, 3, 4, 5, 6, 7])
# self.lfh.write("Assigned data category state-----------------\n")
# aCat.dumpIt(fh=self.lfh)
tmpf = tempname(suffix=".cif")
curContainer.append(aCat)
myDataList.append(curContainer)
ofh = open(tmpf, "w")
pdbxW = PdbxWriter(ofh)
pdbxW.write(myDataList)
ofh.close()
#
#
# Read and update the data -
#
myDataList = []
ifh = open(tmpf, "r")
pRd = PdbxReader(ifh)
pRd.read(myDataList)
ifh.close()
#
myBlock = myDataList[0]
myBlock.printIt()
myCat = myBlock.getObj("pdbx_seqtool_mapping_ref")
myCat.printIt()
for iRow in range(0, myCat.getRowCount()):
myCat.setValue("some value", "ref_mon_id", iRow)
myCat.setValue(100, "ref_mon_num", iRow)
ofh = open(tempname(suffix=".cif"), "w")
pdbxW = PdbxWriter(ofh)
pdbxW.write(myDataList)
ofh.close()
#
except:
traceback.print_exc(file=self.lfh)
self.fail()
def testReadWriteDataFile(self):
"""Test case - data file read write test"""
self.lfh.write(
"\nStarting %s %s\n" %
(self.__class__.__name__, sys._getframe().f_code.co_name))
try:
myDataList = []
ifh = open(self.pathPdbxDataFile, "r")
pRd = PdbxReader(ifh)
pRd.read(myDataList)
ifh.close()
ofh = open(self.pathOutputFile, "w")
pWr = PdbxWriter(ofh)
pWr.write(myDataList)
ofh.close()
except:
traceback.print_exc(file=self.lfh)
self.fail()