def test2(self): directory_name = os.path.dirname(__file__) filename = os.path.join(directory_name, 'plummer128.nemo') file = open(filename, 'rb') nemofile = nemobin.NemoBinaryFile(file) item = nemofile.read_item() self.assertEqual(item.data, "init_xrandom: seed used 123") file.close()
def test7(self): directory_name = os.path.dirname(__file__) filename = os.path.join(directory_name, 'plummer128.nemo') file = open(filename, 'rb') nemofile = nemobin.NemoBinaryFile(file) data = nemofile.read() file.close() outputfile = BytesIO() nemooutputfile = nemobin.NemoBinaryFile(outputfile) nemooutputfile.write(data) string = outputfile.getvalue() outputfile.close() file = open(filename, 'rb') original = file.read() file.close() self.assertEqual(len(original), len(string)) self.assertEqual(original, string)
def test1(self): directory_name = os.path.dirname(__file__) filename = os.path.join(directory_name, 'plummer128.nemo') file = open(filename, 'rb') nemofile = nemobin.NemoBinaryFile(file) tagcharacter, tagstring, dim, mustswap = nemofile.get_item_header() self.assertEqual(tagcharacter, 'c') self.assertEqual(tagstring, 'Headline') self.assertEqual(len(dim), 1) self.assertEqual(dim[0], 28) file.close()
def test5(self): directory_name = os.path.dirname(__file__) filename = os.path.join(directory_name, 'plummer128.nemo') file = open(filename, 'rb') nemofile = nemobin.NemoBinaryFile(file) data = nemofile.read() file.close() outputfile = BytesIO() nemooutputfile = nemobin.NemoBinaryFile(outputfile) nemooutputfile.write(data) string = outputfile.getvalue() outputfile.close() inputfile = BytesIO(string) nemoinputfile = nemobin.NemoBinaryFile(inputfile) tagcharacter, tagstring, dim, mustswap = nemoinputfile.get_item_header() self.assertEqual(tagcharacter, 'c') self.assertEqual(tagstring, 'Headline') self.assertEqual(len(dim), 1) self.assertEqual(dim[0], 28) inputfile.close()
def test3(self): directory_name = os.path.dirname(__file__) filename = os.path.join(directory_name, 'plummer128.nemo') file = open(filename, 'rb') nemofile = nemobin.NemoBinaryFile(file) data = nemofile.read() file.close() self.assertEqual(len(data), 3) tags = list(data.keys()) self.assertEqual(tags[0], 'Headline') self.assertEqual(tags[1], 'History') self.assertEqual(tags[2], 'SnapShot') self.assertEqual(data['History'][0].data, 'mkplummer out=plummer128.nemo nbody=128 seed=123 VERSION=2.8b') self.assertEqual(len(data['SnapShot'][0].data), 2) tags = list(data['SnapShot'][0].data.keys()) self.assertEqual(tags[0], 'Parameters') self.assertEqual(tags[1], 'Particles')
def test6(self): inputfile = BytesIO() nemoinputfile = nemobin.NemoBinaryFile(inputfile) data = nemoinputfile.read() self.assertEqual(len(data), 0)