Source code for lcc.tests.test_connectors

import unittest

from lcc.db_tier.stars_provider import StarsProvider
from lcc.entities.star import Star
import numpy as np
from lcc.utils.stars import plotStarsPicture


[docs]class Test(unittest.TestCase):
[docs] def testKepler(self): # NOTE: Ok RESULTS_NUM = 4 EQUAL = (1, 2) queries = [{"ra": 297.8399, "dec": 46.57427, "delta": 10, "nearest": True}, {"kic_num": 9787239}, {"kic_jkcolor": (0.3, 0.4), "max_records": 2}] client = StarsProvider().getProvider(obtain_method="KeplerArchive", obtain_params=queries) stars = client.getStarsWithCurves() self.failIf(np.NaN in stars[1].lightCurve.getHistogram()[1]) self.failUnless(stars and len(stars) == RESULTS_NUM)
[docs] def testCorotFaint(self): queries = [{"Corot": "102706554"}, {"ra": 100.94235, "dec": -00.89651, "delta": 10}] client = StarsProvider().getProvider( obtain_method="CorotFaintArchive", obtain_params=queries) stars = client.getStarsWithCurves(max_bins=10000)
# plotStarsPicture(stars)
[docs] def testCorotBright(self): RESULTS_NUM = 2 queries = [{"ra": 102.707, "dec": -0.54089, "delta": 10}, {"CoRot": 116}] client = StarsProvider().getProvider( obtain_method="CorotBrightArchive", obtain_params=queries) stars = client.getStarsWithCurves(max_bins=100) self.failUnless(len(stars) == RESULTS_NUM) self.failIf(None in [st.lightCurve for st in stars])
[docs] def testMacho(self): # RESULTS_NUM = 1 queries = [{"Field": 1, "Tile": 3441, "Seqn": 25}] client = StarsProvider().getProvider(obtain_method="MachoDb", obtain_params=queries) stars = client.getStarsWithCurves() self.failUnless(len(stars) == RESULTS_NUM) self.failUnless(isinstance(stars[0], Star))
[docs] def testOgle(self): queries = [{"starid": 2, "field_num": 1, "target": "lmc"}, {"ra": 5.545575 * 15, "dec": -70.55272, "delta": 30}] client = StarsProvider().getProvider(obtain_method="OgleII", obtain_params=queries) stars = client.getStarsWithCurves() print len(stars) plotStarsPicture(stars[1:]) self.failUnless(len(stars) == 2) self.failUnless(isinstance(stars[0], Star))
[docs] def testAsas(self): queries = [{"ASAS": "000030-3937.5"}, {"ra": 10.08, "dec": -39.625, "delta": 20}, {"ra": 0.1251, "dec": -39.6250, "delta": 10}] client = StarsProvider().getProvider(obtain_method="AsasArchive", obtain_params=queries) stars = client.getStarsWithCurves() self.failIf(len(stars) == 0) self.failUnless(isinstance(stars[0], Star)) print " ,".join([st.coo.to_string() for st in stars])
if __name__ == "__main__": #import sys;sys.argv = ['', 'Test.testName'] unittest.main()