def test_get_urls_for_all_time(self): ons_data = OnsDataTester() url_tuples = ons_data._get_urls_for_all_time() assert len(url_tuples) == 90, len(url_tuples) assert url_tuples[0] == [self.ons_url_base + '?lday=1&lmonth=1&lyear=1998&uday=31&umonth=1&uyear=1998', '1998-01', False], url_tuples[0] assert url_tuples[-2] == [self.ons_url_base + '?lday=1&lmonth=5&lyear=2005&uday=31&umonth=5&uyear=2005', '2005-05', False], url_tuples[-2] assert url_tuples[-1] == [self.ons_url_base + '?lday=1&lmonth=6&lyear=2005&uday=31&umonth=6&uyear=2005', '2005-06_incomplete', True], url_tuples[-1]
def test_download(self): url = 'testurl' url_name = 'UrlName' ons_data = OnsDataTester() res = ons_data.download(url, url_name, force_download=False) assert res == self.ons_cache_path + '/ons_data_UrlName', res assert ons_data.files_downloaded == {res: 'testurl'}, res.items()
def test_get_url_recent_via_flexible(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible(days=7) assert res[ 0] == self.ons_url_base + '?lday=14&lmonth=06&lyear=2005&uday=21&umonth=06&uyear=2005', res[ 0] assert res[1] == '7_days_to_2005-06-21', res[1]
def test_get_url_period(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible( start_date=datetime.date(2005, 6, 7), end_date=datetime.date(2005, 6, 9)) assert_equal(res[0], self.ons_url_base + '?lday=07&lmonth=06&lyear=2005&uday=09&umonth=06&uyear=2005') assert_equal(res[1], '2005-06-07_to_2005-06-09')
def test_get_url_month(self): ons_data = OnsDataTester() res = ons_data._get_url_month(12, 2004) assert res[ 0] == self.ons_url_base + '?lday=1&lmonth=12&lyear=2004&uday=31&umonth=12&uyear=2004', res[ 0] assert res[1] == '2004-12', res[1]
def test_get_url(self): ons_data = OnsDataTester() res = ons_data._get_url(31, 12, 2004, 30, 6, 2005) assert res[ 0] == self.ons_url_base + '?lday=31&lmonth=12&lyear=2004&uday=30&umonth=6&uyear=2005', res[ 0] assert res[1] == '2004-12-31_-_2005-6-30', res[1]
def test_get_url_start_date(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible(days=7, start_date=datetime.date(2005, 6, 7)) assert_equal( res[0], self.ons_url_base + '?lday=07&lmonth=06&lyear=2005&uday=14&umonth=06&uyear=2005') assert_equal(res[1], '7_days_from_2005-06-07')
def test_get_url_end_date(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible(days=7, end_date=datetime.date(2005, 6, 15)) assert_equal( res[0], self.ons_url_base + '?lday=08&lmonth=06&lyear=2005&uday=15&umonth=06&uyear=2005') assert_equal(res[1], '7_days_to_2005-06-15')
def test_get_monthly_urls_since(self): ons_data = OnsDataTester() url_tuples = ons_data._get_monthly_urls_since(1999, 11) assert_equal(len(url_tuples), 68) assert url_tuples[0] == [self.ons_url_base + '?lday=1&lmonth=11&lyear=1999&uday=31&umonth=11&uyear=1999', '1999-11', False], url_tuples[0] assert_equal(url_tuples[1][1], '1999-12') assert_equal(url_tuples[2][1], '2000-01') assert_equal(url_tuples[3][1], '2000-02') assert_equal(url_tuples[-1][1], '2005-06_incomplete')
def test_get_monthly_urls_since(self): ons_data = OnsDataTester() url_tuples = ons_data._get_monthly_urls_since(1999, 11) assert_equal(len(url_tuples), 68) assert url_tuples[0] == [ self.ons_url_base + '?lday=1&lmonth=11&lyear=1999&uday=31&umonth=11&uyear=1999', '1999-11', False ], url_tuples[0] assert_equal(url_tuples[1][1], '1999-12') assert_equal(url_tuples[2][1], '2000-01') assert_equal(url_tuples[3][1], '2000-02') assert_equal(url_tuples[-1][1], '2005-06_incomplete')
def _test_import_recent(self): res = OnsDataTester.import_recent(days=7) assert res == 5, res
def test_get_url_start_date(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible( days=7, start_date=datetime.date(2005, 6, 7)) assert_equal(res[0], self.ons_url_base + '?lday=07&lmonth=06&lyear=2005&uday=14&umonth=06&uyear=2005') assert_equal(res[1], '7_days_from_2005-06-07')
def test_get_url_end_date(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible( days=7, end_date=datetime.date(2005, 6, 15)) assert_equal(res[0], self.ons_url_base + '?lday=08&lmonth=06&lyear=2005&uday=15&umonth=06&uyear=2005') assert_equal(res[1], '7_days_to_2005-06-15')
def test_get_url_recent_via_flexible(self): ons_data = OnsDataTester() res = ons_data._get_url_flexible(days=7) assert res[0] == self.ons_url_base + '?lday=14&lmonth=06&lyear=2005&uday=21&umonth=06&uyear=2005', res[0] assert res[1] == '7_days_to_2005-06-21', res[1]
def test_get_url_month(self): ons_data = OnsDataTester() res = ons_data._get_url_month(12, 2004) assert res[0] == self.ons_url_base + '?lday=1&lmonth=12&lyear=2004&uday=31&umonth=12&uyear=2004', res[0] assert res[1] == '2004-12', res[1]
def test_get_url(self): ons_data = OnsDataTester() res = ons_data._get_url(31, 12, 2004, 30, 6, 2005) assert res[0] == self.ons_url_base + '?lday=31&lmonth=12&lyear=2004&uday=30&umonth=6&uyear=2005', res[0] assert res[1] == '2004-12-31_-_2005-6-30', res[1]