def test_get_variables_for_run_default_variables(self): instrument = InstrumentUtils().get_instrument("valid") experiment = Experiment(reference_number=1) reduction_run = ReductionRun(run_number=123, instrument=instrument, experiment=experiment, run_version=1, status=StatusUtils().get_queued()) variables = InstrumentVariablesUtils().get_variables_for_run(reduction_run) self.assertNotEqual(variables, None, "Expecting some variables to be returned") self.assertNotEqual(variables, [], "Expecting some variables to be returned") self.assertTrue(len(variables) > 0, 'Expecting at least 1 variable returned') self.assertEqual(variables[0].experiment_reference, None, "Not expecting experiment_reference") self.assertEqual(variables[0].start_run, 1, "Expecting start run to be 1 but was %s" % variables[0].start_run)
def getReductionRun(with_variables=True): instrument = InstrumentUtils().get_instrument('valid') experiment = Experiment(reference_number=1) experiment.save() reduction_run = ReductionRun(instrument=instrument, run_number=1, experiment=experiment, run_version=0, status=StatusUtils().get_queued(), script=getValidScript('reduce.py')) reduction_run.save() if with_variables: variable = RunVariable(reduction_run=reduction_run,name='test',value='testvalue1',type='text',is_advanced=False) variable.save() reduction_run.run_variables.add(variable) variable = RunVariable(reduction_run=reduction_run,name='advanced_test',value='testvalue2',type='text',is_advanced=True) variable.save() reduction_run.run_variables.add(variable) reduction_run.save() return reduction_run
def createReductionRun(self): instrument = InstrumentUtils().get_instrument("valid") instrument.save() experiment = Experiment(reference_number=1) experiment.save() reduction_run = ReductionRun(run_number=self.run_number, instrument=instrument, experiment=experiment, run_version=1, status=StatusUtils().get_queued(), script=getValidScript('reduce.py')) self.run_number += 1 reduction_run.save() variables = InstrumentVariablesUtils().get_variables_for_run( reduction_run) VariableUtils().save_run_variables(variables, reduction_run) return reduction_run
def test_get_variables_for_run_experiment_reference(self): instrument = InstrumentUtils().get_instrument("valid") variable = InstrumentVariable( instrument=instrument, name="test", value="test", is_advanced=False, type="text", experiment_reference=99999, ) variable.save() experiment = Experiment(reference_number=99999) reduction_run = ReductionRun(run_number=1, instrument=instrument, experiment=experiment, run_version=1, status=StatusUtils().get_queued()) variables = InstrumentVariablesUtils().get_variables_for_run(reduction_run) self.assertNotEqual(variables, None, "Expecting some variables to be returned") self.assertNotEqual(variables, [], "Expecting some variables to be returned") self.assertTrue(len(variables) > 0, 'Expecting at least 1 variable returned') self.assertEqual(variables[0].experiment_reference, 99999, "Expecting instrument variables to match with experiment reference number")