I’m writing a unittest that will be run on my class’s code. For one of the functions they must write, there are two possible return values that they could return and either one is okay for my purposes.
I’ve been using
JavaScript
x
3
1
actual = my_function_call(arg1, arg2)2
self.assertEqual(actual, expected)3
But this doesn’t work for accepting one of two valid return values, so I’ve changed it to:
JavaScript
1
3
1
actual = my_function_call(arg1, arg2)2
self.assertEqual(actual == expected1 or actual == expected2, True)3
Is there a way to do this that isn’t as hacky?
Advertisement
Answer
There exists assertIn test, which you could use in this case as follows
JavaScript
1
2
1
self.assertIn(actual,[expected1,expected2])2
it does check if actual is present in list holding expected1,expected2.