コード例 #1
0
    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'])
コード例 #2
0
    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'])