From e47610bbfdec12ee82e2e39accef95fe840d4438 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Thu, 19 Oct 2017 14:26:47 -0200 Subject: [PATCH 01/13] refactor X to a more meaningful name in election expenses classifier --- .../classifiers/election_expenses_classifier.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py index d6c3299..7056f8e 100644 --- a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py @@ -14,11 +14,11 @@ class ElectionExpensesClassifier(TransformerMixin): Brazilian Federal Revenue category of companies, preceded by its code. """ - def fit(self, X): + def fit(self, dataset): return self - def transform(self, X=None): + def transform(self, dataset=None): return self - def predict(self, X): - return X['legal_entity'] == '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' + def predict(self, dataset): + return dataset['legal_entity'] == '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' From 993bab40e4df323c671c99eec63d366028818a36 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Thu, 19 Oct 2017 14:30:55 -0200 Subject: [PATCH 02/13] Remove a Rails accent of use subject in favor of Zen of Python: explicit is better than implicit and readbility counts --- .../tests/test_election_expenses_classifier.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index 69eea6c..4f58cf3 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -11,16 +11,16 @@ class TestElectionExpensesClassifier(TestCase): def setUp(self): self.dataset = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', dtype={'name': np.str, 'legal_entity': np.str}) - self.subject = ElectionExpensesClassifier() + self.election_expenser_classifier = ElectionExpensesClassifier() def test_is_election_company(self): - self.assertEqual(self.subject.predict(self.dataset)[0], True) + self.assertEqual(self.election_expenser_classifier.predict(self.dataset)[0], True) def test_is_not_election_company(self): - self.assertEqual(self.subject.predict(self.dataset)[1], False) + self.assertEqual(self.election_expenser_classifier.predict(self.dataset)[1], False) def test_fit(self): - self.assertEqual(self.subject.fit(self.dataset), self.subject) + self.assertEqual(self.election_expenser_classifier.fit(self.dataset), self.election_expenser_classifier) def test_tranform(self): - self.assertEqual(self.subject.transform(), self.subject) + self.assertEqual(self.election_expenser_classifier.transform(), self.election_expenser_classifier) From ee324e9cd4116681f2d64658c17d64284c66a187 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Thu, 19 Oct 2017 19:03:28 -0200 Subject: [PATCH 03/13] changes from dataset to dataframe metions in election expenses classifier --- .../classifiers/election_expenses_classifier.py | 8 ++++---- .../tests/test_election_expenses_classifier.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py index 7056f8e..69cc2cf 100644 --- a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py @@ -14,11 +14,11 @@ class ElectionExpensesClassifier(TransformerMixin): Brazilian Federal Revenue category of companies, preceded by its code. """ - def fit(self, dataset): + def fit(self, dataframe): return self - def transform(self, dataset=None): + def transform(self, dataframe=None): return self - def predict(self, dataset): - return dataset['legal_entity'] == '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' + def predict(self, dataframe): + return dataframe['legal_entity'] == '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index 4f58cf3..f55021e 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -9,18 +9,18 @@ class TestElectionExpensesClassifier(TestCase): def setUp(self): - self.dataset = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', + self.dataframe = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', dtype={'name': np.str, 'legal_entity': np.str}) self.election_expenser_classifier = ElectionExpensesClassifier() def test_is_election_company(self): - self.assertEqual(self.election_expenser_classifier.predict(self.dataset)[0], True) + self.assertEqual(self.election_expenser_classifier.predict(self.dataframe)[0], True) def test_is_not_election_company(self): - self.assertEqual(self.election_expenser_classifier.predict(self.dataset)[1], False) + self.assertEqual(self.election_expenser_classifier.predict(self.dataframe)[1], False) def test_fit(self): - self.assertEqual(self.election_expenser_classifier.fit(self.dataset), self.election_expenser_classifier) + self.assertEqual(self.election_expenser_classifier.fit(self.dataframe), self.election_expenser_classifier) def test_tranform(self): self.assertEqual(self.election_expenser_classifier.transform(), self.election_expenser_classifier) From e572e9ba8b2286c76783a9f6a617e57b8c6c8879 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Thu, 19 Oct 2017 19:05:35 -0200 Subject: [PATCH 04/13] Gives a more meaningful constant name for a election legal entity used on election expenses classifier --- .../classifiers/election_expenses_classifier.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py index 69cc2cf..8cf1cd6 100644 --- a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py @@ -21,4 +21,6 @@ def transform(self, dataframe=None): return self def predict(self, dataframe): - return dataframe['legal_entity'] == '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' + ELECTION_LEGAL_ENTITY = '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' + + return dataframe['legal_entity'] == ELECTION_LEGAL_ENTITY \ No newline at end of file From afce56edfb81d456d0f00a162908cd4255bb468b Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Thu, 19 Oct 2017 19:12:14 -0200 Subject: [PATCH 05/13] Makes the assertion more clear about what is been asserted --- .../tests/test_election_expenses_classifier.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index f55021e..4f71e5f 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -14,10 +14,14 @@ def setUp(self): self.election_expenser_classifier = ElectionExpensesClassifier() def test_is_election_company(self): - self.assertEqual(self.election_expenser_classifier.predict(self.dataframe)[0], True) + prediction_result = self.election_expenser_classifier.predict(self.dataframe) + + self.assertEqual(prediction_result[0], True) def test_is_not_election_company(self): - self.assertEqual(self.election_expenser_classifier.predict(self.dataframe)[1], False) + prediction_result = self.election_expenser_classifier.predict(self.dataframe) + + self.assertEqual(prediction_result[1], False) def test_fit(self): self.assertEqual(self.election_expenser_classifier.fit(self.dataframe), self.election_expenser_classifier) From 8fa947eca82d0379eae323c8aca66a7b2a8378e1 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Thu, 19 Oct 2017 19:14:06 -0200 Subject: [PATCH 06/13] fixes a typo on transform for election expense classifier test --- .../tests/test_election_expenses_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index 4f71e5f..e82cd09 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -26,5 +26,5 @@ def test_is_not_election_company(self): def test_fit(self): self.assertEqual(self.election_expenser_classifier.fit(self.dataframe), self.election_expenser_classifier) - def test_tranform(self): + def test_transform(self): self.assertEqual(self.election_expenser_classifier.transform(), self.election_expenser_classifier) From 4ea49746d7d1f1f74675e536118c95cbfc961e61 Mon Sep 17 00:00:00 2001 From: Felipe de Morais Date: Fri, 27 Oct 2017 00:33:15 -0200 Subject: [PATCH 07/13] Gives a more meaningful name for transform test on election expenses classifier --- .../classifiers/election_expenses_classifier.py | 2 +- .../tests/test_election_expenses_classifier.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py index 8cf1cd6..6916f01 100644 --- a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py @@ -18,7 +18,7 @@ def fit(self, dataframe): return self def transform(self, dataframe=None): - return self + pass def predict(self, dataframe): ELECTION_LEGAL_ENTITY = '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index e82cd09..9edf206 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -26,5 +26,5 @@ def test_is_not_election_company(self): def test_fit(self): self.assertEqual(self.election_expenser_classifier.fit(self.dataframe), self.election_expenser_classifier) - def test_transform(self): - self.assertEqual(self.election_expenser_classifier.transform(), self.election_expenser_classifier) + def test_transform_just_for_formality_because_its_never_used(self): + self.assertTrue(self.election_expenser_classifier.transform() is None) From d906fe7feb379b2bfe268eea72e63fad6197993e Mon Sep 17 00:00:00 2001 From: Felipe de Morais Date: Fri, 27 Oct 2017 00:35:52 -0200 Subject: [PATCH 08/13] Gives a more meaningful name for fit test on election expenses classifier --- .../classifiers/election_expenses_classifier.py | 2 +- .../tests/test_election_expenses_classifier.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py index 6916f01..8e9956c 100644 --- a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py @@ -15,7 +15,7 @@ class ElectionExpensesClassifier(TransformerMixin): """ def fit(self, dataframe): - return self + pass def transform(self, dataframe=None): pass diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index 9edf206..ce6c8ab 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -23,8 +23,8 @@ def test_is_not_election_company(self): self.assertEqual(prediction_result[1], False) - def test_fit(self): - self.assertEqual(self.election_expenser_classifier.fit(self.dataframe), self.election_expenser_classifier) + def test_fit_just_for_formality_because_its_never_used(self): + self.assertTrue(self.election_expenser_classifier.fit(self.dataframe) is None) def test_transform_just_for_formality_because_its_never_used(self): self.assertTrue(self.election_expenser_classifier.transform() is None) From 250022a82d531a45808c2c3ab3343d09ce083ded Mon Sep 17 00:00:00 2001 From: Felipe de Morais Date: Fri, 27 Oct 2017 01:05:11 -0200 Subject: [PATCH 09/13] refactor test to just read from fixture file when necessary to improve test performance --- .../tests/test_election_expenses_classifier.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index ce6c8ab..dfae961 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -9,22 +9,27 @@ class TestElectionExpensesClassifier(TestCase): def setUp(self): - self.dataframe = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', - dtype={'name': np.str, 'legal_entity': np.str}) self.election_expenser_classifier = ElectionExpensesClassifier() - def test_is_election_company(self): + def test_legal_entity_is_a_election_company(self): + self.dataframe = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', + dtype={'name': np.str, 'legal_entity': np.str}) + prediction_result = self.election_expenser_classifier.predict(self.dataframe) self.assertEqual(prediction_result[0], True) - def test_is_not_election_company(self): + def test_legal_entity_is_not_election_company(self): + self.dataframe = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', + dtype={'name': np.str, 'legal_entity': np.str}) + prediction_result = self.election_expenser_classifier.predict(self.dataframe) self.assertEqual(prediction_result[1], False) def test_fit_just_for_formality_because_its_never_used(self): - self.assertTrue(self.election_expenser_classifier.fit(self.dataframe) is None) + empty_dataframe = pd.DataFrame() + self.assertTrue(self.election_expenser_classifier.fit(empty_dataframe) is None) def test_transform_just_for_formality_because_its_never_used(self): self.assertTrue(self.election_expenser_classifier.transform() is None) From d63ba5da75ece96bce58645408f41be07495813f Mon Sep 17 00:00:00 2001 From: Felipe de Morais Date: Thu, 2 Nov 2017 00:50:10 -0200 Subject: [PATCH 10/13] Create dataframe focused on each test scenario --- .../fixtures/election_expenses_classifier.csv | 3 --- .../tests/test_election_expenses_classifier.py | 14 +++++++++----- 2 files changed, 9 insertions(+), 8 deletions(-) delete mode 100644 rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv diff --git a/rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv b/rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv deleted file mode 100644 index 77a594f..0000000 --- a/rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv +++ /dev/null @@ -1,3 +0,0 @@ -congressperson_name,name,legal_entity -CARLOS ALBERTO DA SILVA, ELEICAO 2006 CARLOS ALBERTO DA SILVA DEPUTADO,409-0 - CANDIDATO A CARGO POLITICO ELETIVO -PAULO ROGERIO ROSSETO DE MELO,POSTO ROTA 116 DERIVADOS DE PETROLEO LTDA,401-4 - EMPRESA INDIVIDUAL IMOBILIARIA diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index dfae961..734032c 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -12,20 +12,24 @@ def setUp(self): self.election_expenser_classifier = ElectionExpensesClassifier() def test_legal_entity_is_a_election_company(self): - self.dataframe = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', - dtype={'name': np.str, 'legal_entity': np.str}) + self.dataframe = pd.DataFrame( + data=[['CARLOS ALBERTO DA SILVA', 'ELEICAO 2006 CARLOS ALBERTO DA SILVA DEPUTADO', '409-0 - CANDIDATO A CARGO POLITICO ELETIVO']], + columns=['congressperson_name', 'name', 'legal_entity'] + ) prediction_result = self.election_expenser_classifier.predict(self.dataframe) self.assertEqual(prediction_result[0], True) def test_legal_entity_is_not_election_company(self): - self.dataframe = pd.read_csv('rosie/chamber_of_deputies/tests/fixtures/election_expenses_classifier.csv', - dtype={'name': np.str, 'legal_entity': np.str}) + self.dataframe = pd.DataFrame( + data=[['PAULO ROGERIO ROSSETO DE MELO', 'POSTO ROTA 116 DERIVADOS DE PETROLEO LTDA', '401-4 - EMPRESA INDIVIDUAL IMOBILIARIA']], + columns=['congressperson_name', 'name', 'legal_entity'] + ) prediction_result = self.election_expenser_classifier.predict(self.dataframe) - self.assertEqual(prediction_result[1], False) + self.assertEqual(prediction_result[0], False) def test_fit_just_for_formality_because_its_never_used(self): empty_dataframe = pd.DataFrame() From e9c3ac88452b3ae9087540dea42e8f0f75f96cb8 Mon Sep 17 00:00:00 2001 From: Felipe de Morais Date: Thu, 2 Nov 2017 00:55:37 -0200 Subject: [PATCH 11/13] DRY dataframe creation --- .../tests/test_election_expenses_classifier.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index 734032c..bc5143a 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -12,20 +12,14 @@ def setUp(self): self.election_expenser_classifier = ElectionExpensesClassifier() def test_legal_entity_is_a_election_company(self): - self.dataframe = pd.DataFrame( - data=[['CARLOS ALBERTO DA SILVA', 'ELEICAO 2006 CARLOS ALBERTO DA SILVA DEPUTADO', '409-0 - CANDIDATO A CARGO POLITICO ELETIVO']], - columns=['congressperson_name', 'name', 'legal_entity'] - ) + self.dataframe = self._create_dataframe([['CARLOS ALBERTO DA SILVA', 'ELEICAO 2006 CARLOS ALBERTO DA SILVA DEPUTADO', '409-0 - CANDIDATO A CARGO POLITICO ELETIVO']]) prediction_result = self.election_expenser_classifier.predict(self.dataframe) self.assertEqual(prediction_result[0], True) def test_legal_entity_is_not_election_company(self): - self.dataframe = pd.DataFrame( - data=[['PAULO ROGERIO ROSSETO DE MELO', 'POSTO ROTA 116 DERIVADOS DE PETROLEO LTDA', '401-4 - EMPRESA INDIVIDUAL IMOBILIARIA']], - columns=['congressperson_name', 'name', 'legal_entity'] - ) + self.dataframe = self._create_dataframe([['PAULO ROGERIO ROSSETO DE MELO', 'POSTO ROTA 116 DERIVADOS DE PETROLEO LTDA', '401-4 - EMPRESA INDIVIDUAL IMOBILIARIA']]) prediction_result = self.election_expenser_classifier.predict(self.dataframe) @@ -37,3 +31,6 @@ def test_fit_just_for_formality_because_its_never_used(self): def test_transform_just_for_formality_because_its_never_used(self): self.assertTrue(self.election_expenser_classifier.transform() is None) + + def _create_dataframe(self, dataframe_data): + return pd.DataFrame(data=dataframe_data, columns=['congressperson_name', 'name', 'legal_entity']) From a3bd233008e0dd0fc39d86c271d281b0a20d6127 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Fri, 3 Nov 2017 16:58:32 -0200 Subject: [PATCH 12/13] Moves ELECTION_LEGAL_ENTITY constant to a module level --- .../classifiers/election_expenses_classifier.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py index 8e9956c..6ab6225 100644 --- a/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/classifiers/election_expenses_classifier.py @@ -1,6 +1,8 @@ from sklearn.base import TransformerMixin +ELECTION_LEGAL_ENTITY = '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' + class ElectionExpensesClassifier(TransformerMixin): """ Election Expenses classifier. @@ -21,6 +23,4 @@ def transform(self, dataframe=None): pass def predict(self, dataframe): - ELECTION_LEGAL_ENTITY = '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' - - return dataframe['legal_entity'] == ELECTION_LEGAL_ENTITY \ No newline at end of file + return dataframe['legal_entity'] == ELECTION_LEGAL_ENTITY From 8b14e517df9aab0065a379b4a24c65cd2a584ad4 Mon Sep 17 00:00:00 2001 From: Felipe Morais Date: Fri, 3 Nov 2017 16:59:04 -0200 Subject: [PATCH 13/13] Make test data easier to read on test_election_expenses_classifier --- .../tests/test_election_expenses_classifier.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py index bc5143a..31c2590 100644 --- a/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py +++ b/rosie/chamber_of_deputies/tests/test_election_expenses_classifier.py @@ -12,14 +12,22 @@ def setUp(self): self.election_expenser_classifier = ElectionExpensesClassifier() def test_legal_entity_is_a_election_company(self): - self.dataframe = self._create_dataframe([['CARLOS ALBERTO DA SILVA', 'ELEICAO 2006 CARLOS ALBERTO DA SILVA DEPUTADO', '409-0 - CANDIDATO A CARGO POLITICO ELETIVO']]) + self.dataframe = self._create_dataframe([[ + 'CARLOS ALBERTO DA SILVA', + 'ELEICAO 2006 CARLOS ALBERTO DA SILVA DEPUTADO', + '409-0 - CANDIDATO A CARGO POLITICO ELETIVO' + ]]) prediction_result = self.election_expenser_classifier.predict(self.dataframe) self.assertEqual(prediction_result[0], True) def test_legal_entity_is_not_election_company(self): - self.dataframe = self._create_dataframe([['PAULO ROGERIO ROSSETO DE MELO', 'POSTO ROTA 116 DERIVADOS DE PETROLEO LTDA', '401-4 - EMPRESA INDIVIDUAL IMOBILIARIA']]) + self.dataframe = self._create_dataframe([[ + 'PAULO ROGERIO ROSSETO DE MELO', + 'POSTO ROTA 116 DERIVADOS DE PETROLEO LTDA', + '401-4 - EMPRESA INDIVIDUAL IMOBILIARIA' + ]]) prediction_result = self.election_expenser_classifier.predict(self.dataframe)