def test_size_small(self, input_details): # arrange input_details.side_effect = [2] expected = "small" # act actual = d.size() # assert self.assertEqual(expected, actual)
def test_size_large(self, input_details): # arrange input_details.side_effect = [1] expected = "large" # act actual = d.size() # assert self.assertEqual(expected, actual)