Python walk_group_objectsの例

コード例 #1

ファイル: test_ncobj.py プロジェクト: pp-mo/ncobj

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

コード例 #2

ファイル: test_grouping.py プロジェクト: pp-mo/ncobj

 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)