From 72f6afc05850fdde02b4a5d087fe2f299995218f Mon Sep 17 00:00:00 2001 From: Renato Souza Date: Mon, 19 Feb 2024 23:29:58 -0300 Subject: [PATCH 1/3] =?UTF-8?q?Corre=C3=A7=C3=A3o=20de=20erro=20ao=20chama?= =?UTF-8?q?r=20fun=C3=A7=C3=A3o=20replace=20para=20renomear=20as=20colunas?= =?UTF-8?q?=20do=20dataframe?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- statstests/tests.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/statstests/tests.py b/statstests/tests.py index 45dfe97..2e14e75 100644 --- a/statstests/tests.py +++ b/statstests/tests.py @@ -123,9 +123,9 @@ def overdisp(model, data): model.model.data.orig_exog], axis=1) # adjust column names with special characters from categorical columns - df.columns = df.columns.str.replace('[', '', regex=True) - df.columns = df.columns.str.replace('.', '_', regex=True) - df.columns = df.columns.str.replace(']', '', regex=True) + df.columns = df.columns.str.replace('\[', '', regex=True) + df.columns = df.columns.str.replace('\.', '_', regex=True) + df.columns = df.columns.str.replace('\]', '', regex=True) # adjust formula with special characters from categorical columns formula = formula.replace("[", "") From 6beeac8ba37c819eff64905ecfc94c6233088ee8 Mon Sep 17 00:00:00 2001 From: Renato Souza Date: Mon, 19 Feb 2024 23:40:34 -0300 Subject: [PATCH 2/3] =?UTF-8?q?Inclus=C3=A3o=20de=20teste=20(pytest)=20par?= =?UTF-8?q?a=20validar=20altera=C3=A7=C3=A3o?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- statstests/pytests/test.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 statstests/pytests/test.py diff --git a/statstests/pytests/test.py b/statstests/pytests/test.py new file mode 100644 index 0000000..759fdd3 --- /dev/null +++ b/statstests/pytests/test.py @@ -0,0 +1,13 @@ +def test_replace(): + """ + Teste para validar alteração + """ + import pytest + import pandas as pd + df = pd.DataFrame({'[col.1]': [1, 2], '[col.2]': [3, 4]}) + df.columns = df.columns.str.replace('\[', '', regex=True) + df.columns = df.columns.str.replace('\.', '_', regex=True) + df.columns = df.columns.str.replace('\]', '', regex=True) + + df_expected = pd.DataFrame({'col_1': [1, 2], 'col_2': [3, 4]}) + assert df.columns.all() == df_expected.columns.all() From b7907f973e95fd58aa0655308517237b30a92302 Mon Sep 17 00:00:00 2001 From: Renato Souza Date: Sun, 22 Sep 2024 00:42:17 -0300 Subject: [PATCH 3/3] Update test.py --- statstests/pytests/test.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/statstests/pytests/test.py b/statstests/pytests/test.py index 759fdd3..bdc944e 100644 --- a/statstests/pytests/test.py +++ b/statstests/pytests/test.py @@ -1,9 +1,10 @@ +import pytest +import pandas as pd + def test_replace(): """ - Teste para validar alteração - """ - import pytest - import pandas as pd + Test to validate changes + """ df = pd.DataFrame({'[col.1]': [1, 2], '[col.2]': [3, 4]}) df.columns = df.columns.str.replace('\[', '', regex=True) df.columns = df.columns.str.replace('\.', '_', regex=True)