Source code for lcc.tests.test_comp_descriptors

from lcc.stars_processing.descriptors.curve_shape_descr import CurvesShapeDescr
from lcc.entities.star import Star
import numpy as np
import unittest
from lcc.stars_processing.descriptors.hist_shape_descr import HistShapeDescr
from lcc.stars_processing.descriptors.variogram_shape_descr import VariogramShapeDescr


[docs]class TestComparative(unittest.TestCase):
[docs] def setUp(self): self.star1 = Star() x = np.linspace(1, 10, 100) self.star1.lightCurve = [x, np.sin(x)] self.star2 = Star() self.star2.lightCurve = [x, np.cos(x)] self.star3 = Star() self.star3.lightCurve = [x, np.cos(x + 0.5)]
[docs] def testCurveShape(self): lcdes = CurvesShapeDescr(50, 5) lcdes.loadCompStars([self.star2, self.star3]) assert lcdes.getSpaceCoords( [self.star1], "closest")[0] < lcdes.getSpaceCoords([self.star1])[0]
[docs] def testHistShape(self): hist = HistShapeDescr(10, 5) hist.loadCompStars([self.star2, self.star3]) assert hist.getSpaceCoords( [self.star1], "closest")[0] == hist.getSpaceCoords([self.star1])[0]
[docs] def testVarioShape(self): vario = VariogramShapeDescr(10, 5) vario.loadCompStars([self.star2, self.star3]) assert vario.getSpaceCoords( [self.star1]) == vario.getSpaceCoords([self.star1], "closest")
if __name__ == "__main__": unittest.main()