42 lines
1.0 KiB
Python
42 lines
1.0 KiB
Python
|
import unittest
|
||
|
|
from isceobj.Location.Offset import OffsetField, Offset
|
||
|
|
|
||
|
|
class OffsetTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.offsetField = OffsetField()
|
||
|
|
for i in range(10):
|
||
|
|
snr = 1.0
|
||
|
|
if (i == 5):
|
||
|
|
snr = 0.3
|
||
|
|
elif (i == 8):
|
||
|
|
snr = 0.1
|
||
|
|
offset = Offset(x=i,y=i,dx=1,dy=2,snr=snr)
|
||
|
|
self.offsetField.addOffset(offset)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def testCull(self):
|
||
|
|
"""
|
||
|
|
Test that culling offsets below a given signal-to-noise
|
||
|
|
works.
|
||
|
|
"""
|
||
|
|
culledOffsetField = self.offsetField.cull(1.0)
|
||
|
|
i = 0
|
||
|
|
for offset in culledOffsetField:
|
||
|
|
if (offset.getSignalToNoise() < 1.0):
|
||
|
|
self.fail()
|
||
|
|
i = i+1
|
||
|
|
self.assertEquals(i,8)
|
||
|
|
|
||
|
|
def testNaN(self):
|
||
|
|
"""
|
||
|
|
Test that NaN signal-to-noise values are converted to 0.0.
|
||
|
|
"""
|
||
|
|
nanOffset = Offset(x=4,y=5,dx=8,dy=9,snr='nan')
|
||
|
|
self.assertAlmostEquals(nanOffset.getSignalToNoise(),0.0,5)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|