diff --git a/rosie/chamber_of_deputies/__init__.py b/rosie/chamber_of_deputies/__init__.py index 1c600c2..e08e060 100644 --- a/rosie/chamber_of_deputies/__init__.py +++ b/rosie/chamber_of_deputies/__init__.py @@ -3,7 +3,7 @@ import numpy as np from sklearn.externals import joblib -from rosie.chamber_of_deputies.classifiers.dataset import Dataset +from rosie.chamber_of_deputies.dataset import Dataset from rosie.chamber_of_deputies.classifiers.election_expenses_classifier import ElectionExpensesClassifier from rosie.chamber_of_deputies.classifiers.invalid_cnpj_cpf_classifier import InvalidCnpjCpfClassifier from rosie.chamber_of_deputies.classifiers.meal_price_outlier_classifier import MealPriceOutlierClassifier diff --git a/rosie/chamber_of_deputies/__main__.py b/rosie/chamber_of_deputies/__main__.py deleted file mode 100644 index b80c209..0000000 --- a/rosie/chamber_of_deputies/__main__.py +++ /dev/null @@ -1,4 +0,0 @@ -from chamber_of_deputies import main - - -main() \ No newline at end of file diff --git a/rosie/chamber_of_deputies/classifiers/dataset.py b/rosie/chamber_of_deputies/dataset.py similarity index 100% rename from rosie/chamber_of_deputies/classifiers/dataset.py rename to rosie/chamber_of_deputies/dataset.py diff --git a/rosie/chamber_of_deputies/tests/test_dataset.py b/rosie/chamber_of_deputies/tests/test_dataset.py index ec5ed43..eee6919 100644 --- a/rosie/chamber_of_deputies/tests/test_dataset.py +++ b/rosie/chamber_of_deputies/tests/test_dataset.py @@ -4,7 +4,7 @@ from unittest.mock import patch from shutil import copy2 -from rosie.chamber_of_deputies.classifiers.dataset import Dataset +from rosie.chamber_of_deputies.dataset import Dataset class TestDataset(TestCase): @@ -16,8 +16,8 @@ def setUp(self): copy2('rosie/chamber_of_deputies/tests/fixtures/reimbursements.xz', temp_path) self.subject = Dataset(temp_path) - @patch('rosie.chamber_of_deputies.classifiers.dataset.CEAPDataset') - @patch('rosie.chamber_of_deputies.classifiers.dataset.fetch') + @patch('rosie.chamber_of_deputies.dataset.CEAPDataset') + @patch('rosie.chamber_of_deputies.dataset.fetch') def test_get_performs_a_left_merge_between_reimbursements_and_companies(self, _ceap_dataset, _fetch): dataset = self.subject.get() self.assertEqual(5, len(dataset))