Skip to content

Add star-mass-test #90

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
lumi-a wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Add star-mass-test #90

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 13 additions & 1 deletion tests/test_universe.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

from physics import PosVel
from ship import PlayerConfig, PlayerShip
from universe import Planet, PlanetConfig, Universe
from universe import Planet, PlanetConfig, Star, Universe, UniverseOptions


def test_mutual_bounce(monkeypatch: pytest.MonkeyPatch) -> None:
Expand Down Expand Up @@ -180,6 +180,18 @@ def test_gravitational_well() -> None:
)


@pytest.mark.parametrize("small_universe", [True, False])
def test_star_has_majority_of_mass_in_universe(*, small_universe: bool) -> None:
universe, _ = Universe.from_options(UniverseOptions(small=small_universe, splitscreen=True, invincible=False))
star: Star = getattr(universe, "_Universe__star") # Promise me to NEVER do this outside of tests # noqa: B009
star_mass = star.mass
other_mass = sum(
body.mass for body in chain(universe._player_ships, universe._enemy_ships, *universe._planet_chunks.values())
)
total_mass = star_mass + other_mass
assert star.mass / total_mass > 0.99, "Universe's star should contain most of the mass"


def test_planet_generation() -> None:
# Test that planets generated by generate_planet are not too large or too small.

Expand Down