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)