Enhancing tests

rosie.py

@@ -8,8 +8,8 @@
     python rosie.py test [chamber_of_deputies | federal_senate]
 
 Options:
-    --years   runs rosie wuth years filter
-    --path    runs rosie with specific directory
+    --years   runs Rosie fetching data and finding suspicions for a specific set of years
+    --path    directory in which Rosie saves fetched data and the suspicions CSV [default: /tmp/serenata-data]
 
 """
 

rosie/chamber_of_deputies/tests/test_adapter.py

@@ -34,13 +34,13 @@ def test_quantity_of_downloaded_datasets_all_years_fetch(self):
         years = AVAILABLE_YEARS
         chamber_of_deputies = Dataset(self.temp_path, years)
         datasets = chamber_of_deputies.fetch()
-        self.assertEqual(10, len(datasets))
+        self.assertEqual(len(AVAILABLE_YEARS) + 1, len(datasets)) # data from all years + suspicions file
 
     def test_filter_years_fetch(self):
-        years = [2017,2016]
+        years = [2017, 2016]
         chamber_of_deputies = Dataset(self.temp_path, years)
         datasets = chamber_of_deputies.fetch()
-        self.assertEqual(3, len(datasets))
+        self.assertEqual(3, len(datasets)) # data from 2 years + suspicions file
 
     @patch('rosie.chamber_of_deputies.adapter.Dataset')
     @patch('rosie.chamber_of_deputies.adapter.fetch')