def test_asteroid_numbers(_resonancesfixture, start: int, stop: int, fixture_indexes: Tuple[int]): asteroid_nums, planets = _resonancesfixture counter = 0 for resonance in get_resonances(start, stop, False, planets): assert resonance.asteroid_number == asteroid_nums[fixture_indexes[counter]] counter += 1 assert counter == len(fixture_indexes)
def clear_phases(start: int, stop: int, planets: Tuple[str]): conn = engine.connect() resonance_ids = [] redis_logged = False for resonance in get_resonances(start, stop, False, planets): try: REDIS.delete('%s:%i' % (TABLENAME, resonance.id)) except ConnectionError: redis_logged = _log_redis(redis_logged) resonance_ids.append(str(resonance.id)) filename = get_file_name(resonance.id) if os.path.exists(filename): os.remove(filename) conn.execute("DELETE FROM %s WHERE resonance_id = ANY('{%s}'::int[]);" % (TABLENAME, ','.join(resonance_ids)))
def test_planets(_resonance_fixture_different_planets): asteroid_nums, planets = _resonance_fixture_different_planets for resonance in get_resonances(1, 10 ** 5, False, planets): for body, planet in zip(resonance.get_big_bodies(), planets): assert body.name == planet
def test_ordering(_resonancesfixture): asteroid_nums, planets = _resonancesfixture max_asteroid_num = 0 for resonance in get_resonances(1, 10 ** 5, False, planets): assert resonance.asteroid_number >= max_asteroid_num
def test_only_librations(only_librations, resonance_count, _resonance_fixture_different_librations): asteroid_nums, planets, libration = _resonance_fixture_different_librations assert len([x for x in get_resonances(1, 10 ** 5, only_librations, planets)]) == resonance_count