def test_arg_scope(self): myhelper = self.myhelper myhelper2 = self.myhelper2 n = 15 with brew.arg_scope([myhelper], val=n): res = brew.myhelper(self.model) self.assertEqual(n, res) with brew.arg_scope([myhelper, myhelper2], val=n): res1 = brew.myhelper(self.model) res2 = brew.myhelper2(self.model) self.assertEqual([n, n], [res1, res2])
def test_arg_scope_nested(self): myhelper = self.myhelper n = 16 with brew.arg_scope([myhelper], val=-3), \ brew.arg_scope([myhelper], val=-2): with brew.arg_scope([myhelper], val=n): res = brew.myhelper(None) self.assertEqual(n, res) res = brew.myhelper(None) self.assertEqual(res, -2) res = brew.myhelper(None, val=15) self.assertEqual(res, 15)
def test_arg_scope_nested(self): myhelper = self.myhelper n = 16 with brew.arg_scope([myhelper], val=-3), \ brew.arg_scope([myhelper], val=-2): with brew.arg_scope([myhelper], val=n): res = brew.myhelper(self.model) self.assertEqual(n, res) res = brew.myhelper(self.model) self.assertEqual(res, -2) res = brew.myhelper(self.model, val=15) self.model.Validate() self.assertEqual(res, 15)