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
