根据参数调用不同的函数

Question

寻找改善这段代码的建议。基本上我想基于param创建一个通用的验证函数？

def verify(self, param, val):        
        if param == type
            self.assertEqual(self.tm.get.type, val)
        if param == size
            self.assertEqual(self.tm.get.size, val)

Answer 1

你可以使用getattr：

def verify(self, param, val):
    self.assertEqual(getattr(self.tm.get, param), val)