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')