Python find_trait_and_value Examples

Example #1

    def test_find_trait_and_value(self):
        class MyClass(object):
            pass

        class MyHT(HasTraits):
            pass

        obj = MyClass()
        obj.sub = MyClass()
        obj.sub.sub = MyHT()
        obj.sub.csub = Container()
        obj.a = 1
        obj.sub.b = 2
        obj.sub.sub.c = 3
        obj.sub.csub.add('d', Float(4, iotype='in'))
        result = find_trait_and_value(obj, 'sub.sub.c')
        self.assertEqual(result[0].type, 'python')
        self.assertEqual(result[1], 3)
        result = find_trait_and_value(obj, 'sub.csub.d')
        self.assertEqual(result[0].type, 'trait')
        self.assertEqual(result[1], 4)
        try:
            result = find_trait_and_value(obj, 'sub.foo')
        except AttributeError as err:
            self.assertEqual(str(err),
                             "'MyClass' object has no attribute 'foo'")
        else:
            self.fail("expected AttributeError")

Example #2

File: test_container.py Project: cmheath/OpenMDAO-Framework

 def test_find_trait_and_value(self):
     class MyClass(object):
         pass
     class MyHT(HasTraits):
         pass
     
     obj = MyClass()
     obj.sub = MyClass()
     obj.sub.sub = MyHT()
     obj.sub.csub = Container()
     obj.a = 1
     obj.sub.b = 2
     obj.sub.sub.c = 3
     obj.sub.csub.add('d',Float(4, iotype='in'))
     result = find_trait_and_value(obj, 'sub.sub.c')
     self.assertEqual(result[0].type, 'python')
     self.assertEqual(result[1], 3)
     result = find_trait_and_value(obj, 'sub.csub.d')
     self.assertEqual(result[0].type, 'trait')
     self.assertEqual(result[1], 4)
     try:
         result = find_trait_and_value(obj, 'sub.foo')
     except AttributeError as err:
         self.assertEqual(str(err), "'MyClass' object has no attribute 'foo'")
     else:
         self.fail("expected AttributeError")