From c15c84a07b6c22d0dde3a0f6934354ddef9b67b2 Mon Sep 17 00:00:00 2001 From: Ana Schwendler Date: Fri, 5 Jan 2018 13:30:41 -0200 Subject: [PATCH] Enhancing tests --- rosie.py | 4 ++-- rosie/chamber_of_deputies/tests/test_adapter.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/rosie.py b/rosie.py index 5d00cef..5df8e49 100755 --- a/rosie.py +++ b/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] """ diff --git a/rosie/chamber_of_deputies/tests/test_adapter.py b/rosie/chamber_of_deputies/tests/test_adapter.py index 946fb1b..ea68dd3 100644 --- a/rosie/chamber_of_deputies/tests/test_adapter.py +++ b/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')