def test_doesnt_nullify_anything(self): mydf = self.mydf.copy(deep=True) pattern = 'hello there' expected = pd.DataFrame({'field1': ['hello', 'hello1', 'dontstop', 'why', 'HELLO', '3']}) nullify_fields_if_field_matches_pattern(mydf, 'field1', pattern, ['field1'], case_sensitive=False) assert_series_equal(expected['field1'], mydf['field1'])
def test_nullifies_same_field(self): mydf = self.mydf.copy(deep=True) pattern = 'hello' expected = pd.DataFrame({'field1': [np.nan, np.nan, 'dontstop', 'why', np.nan, '3'], 'field2': ['y', 'y', np.nan, '. .', 'z', 'a'], 'field3': ['1dfasdfa!', 1, 'a', '1 . 2', '.|', np.nan]}) nullify_fields_if_field_matches_pattern(mydf, 'field1', pattern, ['field1'], case_sensitive=False) assert_series_equal(expected['field1'], mydf['field1']) assert_series_equal(expected['field2'], mydf['field2']) assert_series_equal(expected['field3'], mydf['field3'])