def test_merge_uvvis(self):
# Case 1 : one value per extinction
uvvis1 = UvvisSpectrum(peaks=[UvvisPeak(value='345')])
uvvis2 = UvvisSpectrum(peaks=[UvvisPeak(extinction='35000')])
uvvis1.merge_uvvis(uvvis2)
gold = {'peaks': [{'value': '345', 'extinction': '35000'}]}
self.assertEqual(gold, uvvis1.serialize())
# Case 2 : One extinction, multiple values
uvvis1 = UvvisSpectrum(
peaks=[UvvisPeak(
value='123'), UvvisPeak(value='456')])
uvvis2 = UvvisSpectrum(peaks=[UvvisPeak(extinction='35000')])
uvvis1.merge_uvvis(uvvis2)
gold = {
'peaks': [{
'extinction': '35000',
'value': '123'
}, {
'extinction': '35000',
'value': '456'
}]
}
self.assertEqual(gold, uvvis1.serialize())
def test_merge_peaks(self):
uvvis = UvvisSpectrum(
peaks=[UvvisPeak(value='123'),
UvvisPeak(extinction='35000')])
uvvis.merge_peaks(0, 1)
gold = {
'peaks': [{
'extinction': '35000',
'value': '123'
}, {
'extinction': '35000'
}]
}
self.assertEqual(gold, uvvis.serialize())
def test_merge_peaks_and_uvvis(self):
""" Tests merge_peaks_and_uvvis correctly merges values from 1 uvvis object containing multiple peaks to
a list of extinctions containing one peak each"""
uvvis_value = UvvisSpectrum(
peaks=[UvvisPeak(
value='123'), UvvisPeak(value='456')])
uvvis_ext = UvvisSpectrum(peaks=[UvvisPeak(extinction='35000')])
uvvis_ext_2 = UvvisSpectrum(peaks=[UvvisPeak(extinction='40000')])
uvvis_value.merge_peaks_and_uvvis([uvvis_ext, uvvis_ext_2], [0, 1])
gold = {
'peaks': [{
'value': '123',
'extinction': '35000'
}, {
'value': '456',
'extinction': '40000'
}]
}
self.assertEqual(gold, uvvis_value.serialize())