def test_add_zero_padding_no_args():
participants = range(1, 6)
expected_result = pd.Series(['001', '002', '003', '004', '005'])
result = add_zero_padding(participants)
assert result.equals(expected_result)
assert result.index.equals(expected_result.index)
def test_add_zero_padding_length_arg():
participants = range(1, 6)
expected_result = pd.Series(['01', '02', '03', '04', '05'])
result = add_zero_padding(participants, length=2)
assert result.equals(expected_result)
assert result.index.equals(expected_result.index)
def test_add_zero_padding_length_too_short():
participants = range(100, 106)
expected_result = pd.Series(['100', '101', '102', '103', '104', '105'])
result = add_zero_padding(participants, length=2)
assert result.equals(expected_result)
assert result.index.equals(expected_result.index)
def test_add_zero_padding_no_series():
participants = range(1, 6)
expected_result = pd.Series(['001', '002', '003', '004', '005'])
result = add_zero_padding(participants, return_series=False)
assert np.array_equal(result, expected_result)
