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)