def test_add_bad_name(self):
bad_names = [
'parent', 'self', 'for', 'if', 'while', 'sin', 'cos', 'tan'
]
for bad in bad_names:
try:
self.root.add(bad, Container())
except Exception as err:
self.assertEqual(str(err),
": '%s' is a reserved or invalid name" % bad)
else:
self.fail("name '%s' should be illegal" % bad)
def test_save_bad_method(self):
# This test exercises handling references to unbound methods defined
# in __main__. Because of this, it only does it's job if this is the
# main module (not run as part of a larger suite in the buildout dir).
output = StringIO.StringIO()
c1 = Container()
c1.unbound_thing = ContainerTestCase.test_save_bad_method
try:
c1.save(output)
except RuntimeError, exc:
msg = ": _pickle_method: <unbound method ContainerTestCase" \
".test_save_bad_method> with module __main__ (None)"
self.assertEqual(str(exc), msg)
def test_save_bad_filename(self):
# TODO: get make_protected_dir() to work on Windows.
if sys.platform == 'win32':
raise nose.SkipTest()
c1 = Container()
directory = make_protected_dir()
path = os.path.join(directory, 'illegal')
try:
c1.save(path)
except IOError, exc:
msg = ": Can't save to '%s': Permission denied" % path
self.assertEqual(str(exc), msg)
def test_set_metadata(self):
c = Container()
c.add_trait('inp', List(range(1000), ddcomp_start=0, ddcomp_end=-1))
self.assertEqual(c.get_metadata('inp', 'ddcomp_start'), 0)
self.assertEqual(c.get_metadata('inp', 'ddcomp_end'), -1)
c.set_metadata('inp', 'ddcomp_start', 10)
c.set_metadata('inp', 'ddcomp_end', 20)
self.assertEqual(c.get_metadata('inp', 'ddcomp_start'), 10)
self.assertEqual(c.get_metadata('inp', 'ddcomp_end'), 20)
assert_raises(self, "c.set_metadata('inp', 'iotype', 'out')",
globals(), locals(), TypeError,
": Can't set iotype on inp, read-only")
def test_iteration2(self):
# Had been skipping some traits of multiple instances.
class Ext(Container):
resources = Dict(iotype='in')
top = Ext()
top.add('ext2', Ext())
sub = top.add('sub', Container())
sub.add('ext3', Ext())
names = [n for n, x in top.items(recurse=True)]
expected = ['ext2', 'ext2.resources', 'resources',
'sub', 'sub.ext3', 'sub.ext3.resources']
self.assertEqual(sorted(names), expected)
def test_get_attributes(self):
c = Container()
c.add_trait('inp', Float(desc='Stuff', low=-200, high=200))
c.set('inp', 42)
attrs = c.get_attributes()
self.assertTrue("Inputs" in attrs.keys())
check = {
'name': 'inp',
'value': 42.0,
'high': 200.0,
'connected': '',
'low': -200.0,
'type': 'float',
'desc': 'Stuff'
}
for key in check.keys():
self.assertEqual(check[key], attrs["Inputs"][0][key])
def test_build_trait(self):
mbc = MyBuilderContainer()
obj_info = [
'f_in', ('f_out_internal', 'f_out', 'out'), 'i_in', 'i_out',
('b_out_internal', 'b_out', 'out', Bool()),
('i_has_metadata', '', dict(iotype='in', low=-1, high=9))
]
create_io_traits(mbc, obj_info)
create_io_traits(mbc, 'foobar')
self.assertTrue(mbc.get_trait('b_out').is_trait_type(Bool))
self.assertTrue(mbc.get_trait('f_out').is_trait_type(Float))
self.assertEqual(mbc.get_trait('f_out').iotype, 'out')
self.assertTrue(mbc.get_trait('i_in').is_trait_type(Int))
self.assertEqual(mbc.get_trait('f_in').iotype, 'in')
self.assertTrue(mbc.get_trait('foobar').is_trait_type(Float))
self.assertEqual(mbc.get_trait('foobar').iotype, 'in')
self.assertTrue(mbc.get_trait('i_has_metadata').is_trait_type(Int))
self.assertEqual(mbc.get_trait('i_has_metadata').iotype, 'in')
self.assertEqual(mbc.get_trait('i_has_metadata').low, -1)
self.assertEqual(mbc.get_trait('i_has_metadata').high, 9)
code = "create_io_traits(mbc, [{}])"
msg = ": create_io_traits cannot add trait {}"
assert_raises(self, code, globals(), locals(), RuntimeError, msg)
code = "create_io_traits(mbc, ('f_in', 'f.in'))"
msg = ": Can't create 'f.in' because it's a dotted pathname"
assert_raises(self, code, globals(), locals(), NameError, msg)
code = "create_io_traits(mbc, ('f2_in', 'f_in'))"
msg = ": Can't create 'f_in' because it already exists"
assert_raises(self, code, globals(), locals(), RuntimeError, msg)
code = "create_io_traits(mbc, 'does_not_exist')"
msg = ": Can't create trait for 'does_not_exist' because it wasn't found"
assert_raises(self, code, globals(), locals(), AttributeError, msg)
cont = Container()
cont.contains = lambda name: True
code = "create_io_traits(cont, 'xyzzy')"
msg = ": build_trait()"
assert_raises(self, code, globals(), locals(), NotImplementedError,
msg)
def setUp(self):
"""This sets up the following hierarchy of Containers:
root
/ \
c1 c2
/ \
c21 c22
/
c221
/
number
"""
self.root = Container()
self.root.add('c1', Container())
self.root.add('c2', Container())
self.root.c2.add('c21', Container())
self.root.c2.add('c22', Container())
self.root.c2.c22.add('c221', Container())
self.root.c2.c22.c221.add('number', Float(3.14, iotype='in'))
def test_default_naming(self):
cont = Container()
cont.add('container1', Container())
cont.add('container2', Container())
self.assertEqual(get_default_name(Container(), cont), 'container3')
self.assertEqual(get_default_name(Container(), None), 'container1')
def test_add_trait_w_subtrait(self):
obj = Container()
obj.add('lst', List([1, 2, 3], iotype='in'))
obj.add('dct', Dict({}, iotype='in'))
def test_pathname(self):
self.root.add('foo', Container())
self.root.foo.add('foochild', Container())
self.assertEqual(self.root.foo.foochild.get_pathname(), 'foo.foochild')
def test_add_non_container(self):
foo = Container()
foo.add('non_container', 'some string')
self.assertEqual(foo.non_container, 'some string')
def test_get_entry_group(self):
class MyClass(object):
pass
self.assertEqual(_get_entry_group(MyClass()), None)
self.assertEqual(_get_entry_group(Container()), 'openmdao.container')
