diff options
Diffstat (limited to 'src/match_up/tests/__init__.py')
| -rw-r--r-- | src/match_up/tests/__init__.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/src/match_up/tests/__init__.py b/src/match_up/tests/__init__.py index 4dbd883..3166746 100644 --- a/src/match_up/tests/__init__.py +++ b/src/match_up/tests/__init__.py @@ -3,17 +3,15 @@ from os.path import isfile from random import randint from pathlib import PurePath from names import get_first_name, get_last_name -from match_up.model import Player, PlayerDatabase +from match_up.model import Player, PlayerDatabase, MIN_SKILL, MAX_SKILL TEST_DATABASE_LOCATION = PurePath("test_storage.json") -MIN_SKILL = 1 -MAX_SKILL = 10 def generate_test_database(total_players: int = 20) -> None: try: if isfile(TEST_DATABASE_LOCATION): - remove_test_database() + remove(TEST_DATABASE_LOCATION) db = PlayerDatabase(TEST_DATABASE_LOCATION) for _ in range(total_players): db.add_player( @@ -38,5 +36,9 @@ def activate_random_players(activate_players: int = 17) -> None: db.close() -def remove_test_database() -> None: - remove(TEST_DATABASE_LOCATION) +def main(): + pass + + +if __name__ == "__main__": + main() |
