Skip to content

Commit 75617c1

Browse files
committed
fix unit test
1 parent d9b7b9a commit 75617c1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit/test_metrics.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,17 +25,17 @@ def test__sort_by_version():
2525
def test__sort_by_version_with_invalid_versions():
2626
# Setup
2727
data = pd.DataFrame({
28-
'version': pd.Series(['2.7.11+', '2.0.0', 'invalid', '3.0', np.nan], dtype='object'),
29-
'name': ['v4', 'v3', 'v2', 'v5', 'v1'],
28+
'version': pd.Series(['2.0.0', 'invalid', '3.0', np.nan], dtype='object'),
29+
'name': ['v3', 'v2', 'v4', 'v1'],
3030
})
3131

3232
# Run
3333
sorted_df = _sort_by_version(data, 'version')
3434

3535
# Assert
36-
expected_versions = ['3.0', '2.7.11+', '2.0.0', 'invalid', np.nan]
36+
expected_versions = ['3.0', '2.0.0', 'invalid', np.nan]
3737
assert sorted_df['version'].tolist() == expected_versions
38-
assert sorted_df['name'].tolist() == ['v5', 'v4', 'v3', 'v2', 'v1']
38+
assert sorted_df['name'].tolist() == ['v4', 'v3', 'v2', 'v1']
3939

4040

4141
def test__sort_by_version_with_mixed_version_formats():

0 commit comments

Comments
 (0)