def test_FileManager():
save_stars_path = "/tmp/test_FileManage/"
path = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir,
"lcc_web", "sample_data", "sample1")
stars = StarsProvider.getProvider("FileManager", {"path": path}).getStars()
assert len(stars) > 0
assert stars[0].lightCurve
saveStars(stars, path=save_stars_path)
stars_loaded = StarsProvider.getProvider("FileManager", {
"path": save_stars_path
}).getStars()
assert stars == stars_loaded
def queryStar(self, query):
stars = StarsProvider.getProvider(self.db_connector, query).getStars()
status = {"found": [], "lc": [], "passed": [], "star_name": []}
passed_info = None
if stars:
stars_with_lc = []
for star in stars:
status["star_name"].append(star.name)
status["found"].append(True)
if star.lightCurve and len(star.lightCurve.mag):
status["lc"].append(True)
stars_with_lc.append(star)
else:
status["lc"].append(False)
# TODO: Support just one filter
# for star_filter in self.stars_filters:
if self.stars_filters:
passed_info = self.stars_filters[0].getAllPredictions(
stars_with_lc,
with_features=self.save_coords,
check_passing=True)
counter = 0
for i in range(len(stars)):
print("passed_info is none", passed_info is None)
if passed_info is None and status["lc"][i]:
status["passed"].append(True)
self.matchOccured(stars[i])
elif status["lc"][i]:
p = passed_info["passed"].values[counter]
status["passed"].append(p)
counter += 1
if p:
self.matchOccured(stars[i])
else:
status["passed"].append(False)
else:
status["found"].append(False)
status["lc"].append(False)
status["passed"].append(False)
status["star_name"].append("Noname")
self.uploadStatus(status, passed_info)
def test_OgleII():
que2 = {"field": "LMC_SC3", "starid": 5}
queries = [{"field": "LMC_SC3", "starid": i} for i in range(1, 5)]
prov = StarsProvider.getProvider("OgleII", queries)
stars = prov.getStars()
