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)
