def test_flush_dataset_correct_data(self):
job_set = Dataset(self.job_set_resources, dataset_name="jobs")
job_set.add_attribute(self.job_id, "job_id", metadata=AttributeType.PRIMARY)
job_set.add_attribute(self.expected_sic_data, "sic", metadata=AttributeType.COMPUTED)
job_set.flush_dataset()
returned_sic_data = job_set.get_attribute("sic")
returned_id_data = job_set.get_attribute("job_id")
self.assert_(ma.allequal(returned_id_data,self.job_id))
self.assert_(ma.allequal(returned_sic_data,self.expected_sic_data))
def test_compute_one_variable_when_asking_for_attribute_that_is_not_in_cache(self):
job_set = Dataset(self.job_set_resources, dataset_name="jobs")
job_set.add_attribute(self.job_id, "job_id", metadata=AttributeType.PRIMARY)
job_set.flush_dataset()
job_id_variable_name = VariableName('opus_core.jobs.attribute_that_does_not_exist')
logger.enable_hidden_error_and_warning_words()
try:
self.assertRaises(StandardError, job_set._compute_one_variable, job_id_variable_name)
finally:
logger.enable_hidden_error_and_warning_words()
def test_err_when_asking_for_attribute_that_is_not_in_cache(self):
job_set = Dataset(self.job_set_resources, dataset_name="jobs")
job_set.add_attribute(self.job_id, "job_id", metadata=AttributeType.PRIMARY)
job_set.flush_dataset()
job_set.get_attribute('job_id')
self.assertRaises(NameError, job_set.get_attribute, 'attribute_that_does_not_exist')
