Exemplo n.º 1
0
        def test_exclude_vars(self):
            import ncobj.grouping as og

            for var in og.walk_group_objects(self.nco, ncobj.Variable):
                print
                print var
                if var.name in ("exc_1", "exc_2"):
                    var.remove()
            print
            print "new result:"
            print "\n".join(str(x) for x in og.walk_group_objects(self.nco))
Exemplo n.º 2
0
 def test_basic(self):
     walk_objs = list(walk_group_objects(self.root))
     # NOTE: can't use set-equality here, as it test 'is' not '__eq__'.
     for elem in self.all_objs:
         self.assertIn(elem, walk_objs)
     for obj in walk_objs:
         self.assertIn(obj, self.all_objs)