def test_read_comment(self):
comment = random_comment()
num_atoms = random.randint(1,100)
crds = random_crds(num_atoms)
with tfu.TempfileSession() as tfs:
file_name = tfs.temp_file_name('.rst')
with rst.open(file_name, 'w', comment=comment) as f:
f.write(crds)
with rst.open(file_name, 'r') as f:
self.assertEqual(f.comment, comment)
def test_read_comment(self):
with open(test_file('expected_cpr_comment')) as f:
expected_comment = f.next()[:-1]
with rst.open(test_file('cpr')) as f:
self.assertEqual(f.comment, expected_comment)
def test_read_crd(self):
with open(test_file('expected_cpr_crds')) as f:
expected_crds = [float(x) for x in f.read().split()]
with rst.open(test_file('cpr')) as f:
crds = f.next()
for (expected, crd) in it.izip_longest(expected_crds, crds):
self.assertAlmostEqual(expected, crd, 3)
def test_read_crds(self):
comment = random_comment()
num_atoms = random.randint(1,100)
num_crds = 1 # random.randint(1, 10)
crdss = [random_crds(num_atoms) for count in xrange(num_crds)]
with tfu.TempfileSession() as tfs:
file_name = tfs.temp_file_name('.rst')
with rst.open(file_name, 'w', comment=comment) as f:
for crds in crdss:
f.write(crds)
with rst.open(file_name, 'r') as f:
for expected_crds, read_crds in it.izip_longest(crdss, f):
self.assertNotEqual(expected_crds, None)
self.assertNotEqual(read_crds, None)
for (expected, crd) in it.izip_longest(expected_crds, read_crds):
self.assertAlmostEqual(expected, crd, 3)
