class TestOutputWriteAttribute(unittest.TestCase): def setUp(self): self.filename = 'test-Output.h5' self.int_attr_name = 'Dummy Int Attribute' self.double_attr_name = 'Dummy Double Attribute' self.str_attr_name = 'Dummy String Attribute' self.fh = Output(self.filename,'a') def tearDown(self): if os.path.exists(self.filename): self.fh.close() os.remove(self.filename) def test_string_attribute(self): import random str_len = random.randint(16,1024) self.fh.set_attribute(self.str_attr_name,os.urandom(str_len)) def test_double_attribute(self): import random self.fh.set_attribute(self.double_attr_name,random.random()) def test_int_attribute(self): import random self.fh.set_attribute(self.int_attr_name,random.randint(0,100000))
class TestOutputWriteAttribute(unittest.TestCase): def setUp(self): self.filename = 'test-Output.h5' self.int_attr_name = 'Dummy Int Attribute' self.double_attr_name = 'Dummy Double Attribute' self.str_attr_name = 'Dummy String Attribute' self.fh = Output(self.filename, 'a') def tearDown(self): if os.path.exists(self.filename): self.fh.close() os.remove(self.filename) def test_string_attribute(self): import random str_len = random.randint(16, 1024) self.fh.set_attribute(self.str_attr_name, os.urandom(str_len)) def test_double_attribute(self): import random self.fh.set_attribute(self.double_attr_name, random.random()) def test_int_attribute(self): import random self.fh.set_attribute(self.int_attr_name, random.randint(0, 100000))
class TestOutputReadAttribute(unittest.TestCase): def setUp(self): import random import string self.filename = 'test-Output.h5' self.fh = Output(self.filename, 'w') self.attributes = [] self.int_attr_name = 'Dummy Int Attribute' self.attributes.append(self.int_attr_name) self.int_attr = random.randint(0, 1000000) self.fh.set_attribute(self.int_attr_name, self.int_attr) self.double_attr_name = 'Dummy Double Attribute' self.attributes.append(self.double_attr_name) self.double_attr = random.random() self.fh.set_attribute(self.double_attr_name, self.double_attr) self.str_attr_name = 'Dummy String Attribute' self.attributes.append(self.str_attr_name) str_len = random.randint(10, 1024) self.str_attr = ''.join( random.choice(string.letters) for i in xrange(str_len)) self.fh.set_attribute(self.str_attr_name, self.str_attr) def tearDown(self): if os.path.exists(self.filename): self.fh.close() os.remove(self.filename) def test_string_attribute(self): test_str = self.fh.get_attribute(self.str_attr_name) self.assertEqual(test_str, self.str_attr) def test_double_attribute(self): test_double = self.fh.get_attribute(self.double_attr_name) self.assertEqual(test_double, self.double_attr) def test_int_attribute(self): test_int = self.fh.get_attribute(self.int_attr_name) self.assertEqual(test_int, self.int_attr) def test_attributes(self): attributes = self.fh.attributes() for attr in self.attributes: if attr not in attributes: TestCase.fail('Failed to read attributes correctly')
class TestOutputReadAttribute(unittest.TestCase): def setUp(self): import random import string self.filename = 'test-Output.h5' self.fh = Output(self.filename,'w') self.attributes = [] self.int_attr_name = 'Dummy Int Attribute' self.attributes.append(self.int_attr_name) self.int_attr = random.randint(0,1000000) self.fh.set_attribute(self.int_attr_name,self.int_attr) self.double_attr_name = 'Dummy Double Attribute' self.attributes.append(self.double_attr_name) self.double_attr = random.random() self.fh.set_attribute(self.double_attr_name,self.double_attr) self.str_attr_name = 'Dummy String Attribute' self.attributes.append(self.str_attr_name) str_len = random.randint(10,1024) self.str_attr = ''.join(random.choice(string.letters) for i in xrange(str_len)) self.fh.set_attribute(self.str_attr_name,self.str_attr) def tearDown(self): if os.path.exists(self.filename): self.fh.close() os.remove(self.filename) def test_string_attribute(self): test_str = self.fh.get_attribute(self.str_attr_name) self.assertEqual(test_str,self.str_attr) def test_double_attribute(self): test_double = self.fh.get_attribute(self.double_attr_name) self.assertEqual(test_double,self.double_attr) def test_int_attribute(self): test_int = self.fh.get_attribute(self.int_attr_name) self.assertEqual(test_int,self.int_attr) def test_attributes(self): attributes = self.fh.attributes() for attr in self.attributes: if attr not in attributes: TestCase.fail('Failed to read attributes correctly')