def test_get_return(self): fn = FunctionRef("baz", 1, "bazFn") argref, returntype = fn.get_return() self.assertIsNone(argref) self.assertEqual(returntype, types.NoneType) retval = ArgRef(fn, "") retval.add_sample(1) argref, returntype = fn.get_return() self.assertIs(argref, ArgRef(fn, "")) self.assertIs(returntype, int)