def test_abstract_workload_get_default(side_effect, default):
mock_workload = mock.MagicMock(spec=AbstractWorkload)
mock_workload.sets = mock.MagicMock()
mock_workload.sets.get.side_effect = side_effect
ret_val = AbstractWorkload.get(mock_workload, 1, default=default)
assert ret_val == default
def test_abstract_workload_get():
idx = 3
mock_workload = mock.MagicMock(spec=AbstractWorkload)
mock_workload.sets = mock.MagicMock()
mock_workload.sets.get.side_effect = lambda order=None: list(range(5))[
order]
ret_val = AbstractWorkload.get(mock_workload, idx)
assert ret_val == idx