def test_arity(self): # e.g., def square(n): # return n * n # arity(square) = 1 fn = FunctionRef("", 1, "square") squarearg = ArgRef(fn, "n") self.assertEqual(fn.arity(), 1)