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')
