Skip to content

Commit

Permalink
added test_clean_pandas_column_names
Browse files Browse the repository at this point in the history
  • Loading branch information
wverastegui committed Feb 9, 2024
1 parent a83cbf5 commit 8c5e933
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion tests/test_data_PandasData.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,4 +106,14 @@ def test_has_retention_times(filename, expected):
def test_has_retention_indices(filename, expected):
filepath = os.path.join(testdata_dir, filename)
data = PandasDataBuilder().with_filename(filepath).build()
assert data.has_retention_indices() == expected
assert data.has_retention_indices() == expected

def test_clean_pandas_column_names():
#arrange
expected = ["rt"]
builder = PandasDataBuilder().with_filename(os.path.join(testdata_dir, "minimal_unclean_colnames.csv"))
#act
actual = list(builder.build()._data.columns)
#assert
assert set(expected) <= set(actual)

0 comments on commit 8c5e933

Please sign in to comment.