def test_no_bkg(self):
data = self.data
# No info at all
mi = spot.MomentOfInertia(data)
self.assertAlmostEqual(mi, 112.005654085, delta=10)
# now with MD_BASELINE
data.metadata[model.MD_BASELINE] = 100
mi = spot.MomentOfInertia(data)
self.assertAlmostEqual(mi, 112.005654085, delta=5)
def test_spot(self):
data = numpy.zeros((480, 640), dtype=numpy.uint16)
data[240, 360] = 5000
mi = spot.MomentOfInertia(data)
self.assertTrue(math.isnan(mi) or mi > 0)
def test_black(self):
data = numpy.zeros((480, 640), dtype=numpy.uint16)
mi = spot.MomentOfInertia(data)
self.assertTrue(math.isnan(mi))
def test_precomputed(self):
data = self.data
background = self.background
mi = spot.MomentOfInertia(data, background)
self.assertAlmostEqual(mi, 112.005654085)
