Skip to content

Commit

Permalink
Generate: fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Berserker66 committed Jun 12, 2024
1 parent 441db0b commit 1b7436d
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
2 changes: 1 addition & 1 deletion Generate.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def get_seed_name(random_source) -> str:
return f"{random_source.randint(0, pow(10, seeddigits) - 1)}".zfill(seeddigits)


def main(args=None):
def main(args=None) -> Tuple[argparse.Namespace, int]:
if not args:
args = mystery_argparse()

Expand Down
7 changes: 4 additions & 3 deletions test/programs/test_generate.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
from tempfile import TemporaryDirectory

import Generate
import Main


class TestGenerateMain(unittest.TestCase):
Expand Down Expand Up @@ -58,7 +59,7 @@ def test_generate_absolute(self):
'--player_files_path', str(self.abs_input_dir),
'--outputpath', self.output_tempdir.name]
print(f'Testing Generate.py {sys.argv} in {os.getcwd()}')
Generate.main()
Main.main(*Generate.main())

self.assertOutput(self.output_tempdir.name)

Expand All @@ -67,7 +68,7 @@ def test_generate_relative(self):
'--player_files_path', str(self.rel_input_dir),
'--outputpath', self.output_tempdir.name]
print(f'Testing Generate.py {sys.argv} in {os.getcwd()}')
Generate.main()
Main.main(*Generate.main())

self.assertOutput(self.output_tempdir.name)

Expand All @@ -86,7 +87,7 @@ def test_generate_yaml(self):
sys.argv = [sys.argv[0], '--seed', '0',
'--outputpath', self.output_tempdir.name]
print(f'Testing Generate.py {sys.argv} in {os.getcwd()}, player_files_path={self.yaml_input_dir}')
Generate.main()
Main.main(*Generate.main())
finally:
user_path.cached_path = user_path_backup

Expand Down

0 comments on commit 1b7436d

Please sign in to comment.