diff --git a/microsetta_private_api/repo/metadata_repo/_repo.py b/microsetta_private_api/repo/metadata_repo/_repo.py index 25bfbe2d..995cfaae 100644 --- a/microsetta_private_api/repo/metadata_repo/_repo.py +++ b/microsetta_private_api/repo/metadata_repo/_repo.py @@ -74,7 +74,7 @@ def drop_private_columns(df): # sensitive in nature pm_remove = {c.lower() for c in df.columns if c.lower().startswith('pm_')} - freetext_fields = {c.lower () for c in _get_freetext_fields()} + freetext_fields = {c.lower() for c in _get_freetext_fields()} remove = pm_remove | {c.lower() for c in EBI_REMOVE} | freetext_fields to_drop = [c for c in df.columns if c.lower() in remove] @@ -640,7 +640,7 @@ def _find_duplicates(barcodes): def _get_freetext_fields(): """ Retrieve a list of all free-text survey fields from the database - + Returns ------- list of str diff --git a/microsetta_private_api/repo/metadata_repo/tests/test_repo.py b/microsetta_private_api/repo/metadata_repo/tests/test_repo.py index 01e3ead1..81311ba1 100644 --- a/microsetta_private_api/repo/metadata_repo/tests/test_repo.py +++ b/microsetta_private_api/repo/metadata_repo/tests/test_repo.py @@ -332,7 +332,7 @@ def test_drop_private_columns(self): obs = drop_private_columns(df) pdt.assert_frame_equal(obs, exp) - def test_drop_private_columns(self): + def test_drop_private_columns_freetext(self): # This test specifically asserts that the new code to drop free-text # fields works, even if those fields are not represented in the # EBI_REMOVE list