def test_get_num_samples(self):
fn = FunctionRef("bar", 1, "barFn")
arg1 = ArgRef(fn, "arg1")
retval = ArgRef(fn, "")
# Pretend we observed an input, but an exception was thrown.
arg1.add_sample(True)
self.assertEqual(fn.get_num_samples(), 1)
# Observing an output shouldn't change anything
retval.add_sample("asdf")
self.assertEqual(fn.get_num_samples(), 1)
# Observing a successful run should increment by one.
arg1.add_sample(False)
retval.add_sample("foo")
self.assertEqual(fn.get_num_samples(), 2)
