def test_merge(self):
"""
Can we merge multiple ModelJobs.
n.b. Currently this only supports time signals!
TODO: Merge the non-time-signal data.
"""
# n.b. non-zero values. Zero time signals are ignored.
kb_read1 = TimeSignal.from_values('kb_read', [0.0], [1.0], priority=8)
kb_read2 = TimeSignal.from_values('kb_read', [0.0], [1.0], priority=10)
kb_write1 = TimeSignal.from_values('kb_write', [0.0], [1.0],
priority=8)
# Test that we take the union of the available time series
job1 = ModelJob(label="label1", timesignals={'kb_read': kb_read1})
job2 = ModelJob(label="label1", timesignals={'kb_write': kb_write1})
job1.merge(job2)
self.assertEqual(len(job1.timesignals), len(signal_types))
self.assertEqual(job1.timesignals['kb_read'], kb_read1)
self.assertEqual(job1.timesignals['kb_write'], kb_write1)
# (The other time signals should still be None)
for ts_name in signal_types:
if ts_name in ['kb_read', 'kb_write']:
continue
self.assertIn(ts_name, job1.timesignals)
self.assertIsNone(job1.timesignals[ts_name])
# check that when merging we take the signal with highest priority index
job1 = ModelJob(label="label1", timesignals={'kb_read': kb_read1})
job2 = ModelJob(label="label1", timesignals={'kb_read': kb_read2})
job1.merge(job2)
self.assertEqual(job1.timesignals['kb_read'], kb_read2)
def test_merge_ignores_empty_timesignals(self):
"""
When merging in time signals from another job, if there is no data in the "other" time signal, then it
should be ignored for merging purposes.
:return:
"""
kb_read = TimeSignal.from_values('kb_read', [0.0], [1.0])
kb_write = TimeSignal.from_values('kb_write', [0.0],
[0.0]) # n.b. zero data
job1 = ModelJob(label="label1", timesignals={'kb_read': kb_read})
job2 = ModelJob(label="label1", timesignals={'kb_write': kb_write})
self.assertIsNone(job1.timesignals['kb_write'])
self.assertIsNotNone(job2.timesignals['kb_write'])
job1.merge(job2)
self.assertIsNone(job1.timesignals['kb_write'])
def test_merge_fails_different_label(self):
"""
We should not be able to merge two jobs with differing labels, as these don't correspond to the same overall job
"""
job1 = ModelJob(label="a-label-1")
job2 = ModelJob(label="a-label-2")
self.assertRaises(AssertionError, lambda: job1.merge(job2))
def test_merge_rejects_mislabelled_time_signals(self):
"""
Test that the merging routine checks the labelling validity. Both ways around.
:return:
"""
kb_read = TimeSignal.from_values('kb_read', [0.0], [1.0])
kb_write = TimeSignal.from_values('kb_read', [0.0],
[1.0]) # n.b. mislabelled
job1 = ModelJob(label="label1", timesignals={'kb_read': kb_read})
job2 = ModelJob(label="label1")
job2.timesignals['kb_write'] = kb_write
self.assertRaises(ModellingError, lambda: job1.merge(job2))
# And the other way around
job2 = ModelJob(label="label1", timesignals={'kb_read': kb_read})
job1 = ModelJob(label="label1")
job1.timesignals['kb_write'] = kb_write
self.assertRaises(ModellingError, lambda: job1.merge(job2))