Python Unittest with different inputs

Question

I want to apply the same unittest to different inputs and outputs. For instance:

def test_something(self):
   calculated = some_funtion(input_data[idx])
   self.assertEquals(calculated, expected[idx])

for all idx (say, 1,000).

There's this question that asks something similar, but the question does not have any satisfactory answers.

I have to use Python unittest.

score 1 · Accepted Answer · answered Feb 20 '21 at 19:49

If you must stick with unittest and not use the superior pytest, I would recomend to use the parameterized library.

from parameterized import parameterized

class MyTest(unittest.TestCase):
   @parameterized.expand([1,2,3,...., 100])  # your idxs
   def test_something(self, idx):
       calculated = some_funtion(input_data[idx])
       self.assertEquals(calculated, expected[idx])

Python Unittest with different inputs

1 Answers1