Move dataset.py file in order to separate classifiers, tests and sett…

…ings from modules
okfn-brasil · Apr 14, 2017 · 7250d12 · 7250d12
1 parent 372b252
commit 7250d12
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 8 deletions.
diff --git 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
diff --git a/...hamber_of_deputies/classifiers/dataset.py → rosie/chamber_of_deputies/dataset.py b/...hamber_of_deputies/classifiers/dataset.py → 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
@@ -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))