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()