Source code for lcc.tests.test_abbe_descriptor

from matplotlib import pyplot
import unittest

from lcc.entities.star import Star
import numpy as np

# TODO


[docs]class Test(unittest.TestCase):
[docs] def setUp(self): N = 100 x = np.linspace(0, 10, 1000) self.variables = [] for ii in range(N): st = Star(name="VariableStar%i" % ii) st.putLightCurve([x, np.sin(x) + np.random.normal(x) * 0.1]) self.variables.append(st) self.noisy = [] for ii in range(N): st = Star(name="VariableStar%i" % ii) st.putLightCurve([x, np.random.normal(x) * 2]) self.noisy.append(st)
[docs] def testAbbe(self): N = 1000 var_abbes = [st.lightCurve.getAbbe(bins=N) for st in self.variables] noisy_abbes = [st.lightCurve.getAbbe(bins=N) for st in self.noisy] print np.mean(var_abbes) print np.mean(noisy_abbes) pyplot.hist(var_abbes, color="b") pyplot.hist(noisy_abbes, color="r") pyplot.show()
if __name__ == "__main__": #import sys;sys.argv = ['', 'Test.testAbbe'] unittest.main()