class ResultSet_test(unittest.TestCase):
def setUp(self):
self.theSet = ResultSet({'frequency':float,'transmittedPower':Power,'generatedPower':Power})
self.theSet.append({'frequency':30e3,'transmittedPower':Power(30.,'dBm'),'generatedPower':Power(40.,'dBm')})
self.theSet.append({'frequency':40e3,'generatedPower':Power(40.,'dBm')})
def test_floats(self):
self.assertTrue(all(self.theSet['frequency'] == numpy.array([30.0e3,40.0e3])))
def test_numpySubClasses(self):
self.assertTrue( (self.theSet['generatedPower'] == Power([40.,40.],'dBm')).all())
def test_singleValue(self):
self.assertTrue(self.theSet['transmittedPower'][0] == Power([1.],'W'))
def test_nanConversion(self):
self.assertTrue(numpy.isnan(self.theSet['transmittedPower'][1]))
def test_byRow(self):
rows = self.theSet.byRow()
row = rows.next()
self.assertEqual(row['frequency'] , 30e3 )
self.assertEqual( row['transmittedPower'] , Power(30.,'dBm') )
self.assertEqual( row['generatedPower'] , Power(40.,'dBm') )
row = rows.next()
self.assertEqual( row['frequency'] , 40e3 )
self.assertTrue( numpy.isnan(row['transmittedPower']) )
self.assertEqual( row['generatedPower'] , Power(40.,'dBm') )
class ResultSetXml_test(XmlTest):
def setUp(self):
XmlTest.setUp(self)
self.result = ResultSet({'frequency':float,'transmittedPower':Power,'generatedPower':Power})
self.result.append({'frequency':30e3,'transmittedPower':Power(30.,'dBm'),'generatedPower':Power(40.,'dBm')})
self.result.append({'frequency':40e3,'generatedPower':Power(40.,'dBm')})
def test_loopthrough(self):
self.assertLoopThrough(ResultSet)