def test_timestamp(self):
dataset = get_sample_dataset(day=date(2017, 4, 3))
self.assertEqual(
dataset.timestamp,
timezone.make_aware(datetime(2017, 4, 3, 0, 0, 0),
timezone.get_current_timezone()),
)
def test_datafile(self):
dataset = get_sample_dataset(day=date(2017, 4, 3),
datasets={
'20170403_something.txt': 'testing',
})
f = dataset.datafile('something')
self.assertEqual(f.read(), 'testing')
def test_datafile(self):
dataset = get_sample_dataset(
day=date(2017, 4, 3),
datasets={"20170403_something.txt": "testing"},
)
f = dataset.datafile("something")
self.assertEqual(f.read(), b"testing")
def test_validate_file_single_scenario(self):
scenario_file = ContentFile(json.dumps(self.scenario))
dataset = get_sample_dataset()
with patch(
'ratechecker.validation.get_rates',
return_value={'data': {'3.25': '1.75'}}
):
self.validator.validate_file(scenario_file, dataset, scenario_id=0)
def test_timestamp(self):
dataset = get_sample_dataset(day=date(2017, 4, 3))
self.assertEqual(
dataset.timestamp,
timezone.make_aware(
datetime(2017, 4, 3, 0, 0, 0),
timezone.get_current_timezone()
)
)
def test_load_fails_if_dataset_missing(self):
dataset = get_sample_dataset(day=date(2017, 4, 3),
datasets={"20170403_key.txt": "testing"})
loader = Mock()
loader.return_value = loader
dataset.loaders = {"other_key": loader}
with self.assertRaises(KeyError):
dataset.load()
def test_load_calls_loaders(self):
dataset = get_sample_dataset(day=date(2017, 4, 3),
datasets={"20170403_key.txt": "testing"})
loader = Mock()
loader.return_value = loader
dataset.loaders = {"key": loader}
dataset.load()
loader.load.assert_called_once()
def test_validate_file_scenario_fails(self):
scenario_file = ContentFile(json.dumps(self.scenario))
dataset = get_sample_dataset()
with patch(
'ratechecker.validation.get_rates',
return_value={'data': {'3.25': '0.00'}}
):
with self.assertRaises(ValidationError):
self.validator.validate_file(scenario_file, dataset)
def test_datafile(self):
dataset = get_sample_dataset(
day=date(2017, 4, 3),
datasets={
'20170403_something.txt': 'testing',
}
)
f = dataset.datafile('something')
self.assertEqual(f.read(), 'testing')
def test_validate_file_single_scenario(self):
scenario_file = ContentFile(json.dumps(self.scenario))
dataset = get_sample_dataset()
with patch(
"ratechecker.validation.get_rates",
return_value={"data": {
"3.25": "1.75"
}},
):
self.validator.validate_file(scenario_file, dataset, scenario_id=0)
def test_validate_file_scenario_fails(self):
scenario_file = ContentFile(json.dumps(self.scenario))
dataset = get_sample_dataset()
with patch(
"ratechecker.validation.get_rates",
return_value={"data": {
"3.25": "0.00"
}},
):
with self.assertRaises(ValidationError):
self.validator.validate_file(scenario_file, dataset)
def test_load_calls_loaders(self):
dataset = get_sample_dataset(
day=date(2017, 4, 3),
datasets={
'20170403_key.txt': 'testing',
}
)
loader = Mock()
loader.return_value = loader
dataset.loaders = {'key': loader}
dataset.load()
loader.load.assert_called_once()
def test_cover_sheet(self):
dataset = get_sample_dataset()
self.assertIsInstance(dataset.cover_sheet, CoverSheet)
def test_cover_sheet(self):
dataset = get_sample_dataset()
self.assertIsInstance(dataset.cover_sheet, CoverSheet)
