def test_coefficient_updates(self):
sim = SimulatedUSB()
coeff_dict = {'C0': '7.25405E+02',
'C1': '1.43880E-01',
'C2': '7.16617E-06',
'C3': '-8.68137E-09'
}
# Raises error when trying to update without assignment
self.assertRaises(ValueError, sim.new_coefficients, coeff_dict)
self.assertTrue(sim.assign("Stroker785M"))
# Assign new coefficients
self.assertTrue(sim.new_coefficients(coeff_dict))
wavenum_axis, intensity_data = sim.get_line_wavenumber()
first_conv = "-1046.55"
last_conv = "2487.62"
self.assertEqual("%05.2f" % wavenum_axis[0], first_conv)
self.assertEqual("%05.2f" % wavenum_axis[-1], last_conv)
self.assertEqual(intensity_data[0], 0)
self.assertEqual(intensity_data[2047], 2047)
def test_wavenumber_conversion(self):
# Set some default coefficients, make sure the returned
# wavenumber conversion is accurate
sim = SimulatedUSB()
self.assertRaises(ValueError, sim.get_line_wavenumber)
self.assertTrue(sim.assign("Stroker785L"))
self.assertEqual(sim.pixel_count, 1024)
wavenum_axis, intensity_data = sim.get_line_wavenumber()
self.assertEqual(len(intensity_data), 1024)
self.assertEqual(len(wavenum_axis), 1024)
first_conv = "164.33"
last_conv = "836.76"
self.assertEqual("%05.2f" % wavenum_axis[0], first_conv)
self.assertEqual("%05.2f" % wavenum_axis[-1], last_conv)
self.assertEqual(intensity_data[0], 0)
self.assertEqual(intensity_data[1023], 1023)
