mirror of
https://github.com/MarioSpore/Grinch-AP.git
synced 2025-10-21 12:11:33 -06:00
Generate: split ERmain out of main (#3515)
This commit is contained in:
@@ -26,6 +26,7 @@ def _generate_local_inner(games: Iterable[str],
|
||||
with TemporaryDirectory() as players_dir:
|
||||
with TemporaryDirectory() as output_dir:
|
||||
import Generate
|
||||
import Main
|
||||
|
||||
for n, game in enumerate(games, 1):
|
||||
player_path = Path(players_dir) / f"{n}.yaml"
|
||||
@@ -42,7 +43,7 @@ def _generate_local_inner(games: Iterable[str],
|
||||
sys.argv = [sys.argv[0], "--seed", str(hash(tuple(games))),
|
||||
"--player_files_path", players_dir,
|
||||
"--outputpath", output_dir]
|
||||
Generate.main()
|
||||
Main.main(*Generate.main())
|
||||
output_files = list(Path(output_dir).glob('*.zip'))
|
||||
assert len(output_files) == 1
|
||||
final_file = dest / output_files[0].name
|
||||
|
@@ -9,6 +9,7 @@ from pathlib import Path
|
||||
from tempfile import TemporaryDirectory
|
||||
|
||||
import Generate
|
||||
import Main
|
||||
|
||||
|
||||
class TestGenerateMain(unittest.TestCase):
|
||||
@@ -58,7 +59,7 @@ class TestGenerateMain(unittest.TestCase):
|
||||
'--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)
|
||||
|
||||
@@ -67,7 +68,7 @@ class TestGenerateMain(unittest.TestCase):
|
||||
'--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)
|
||||
|
||||
@@ -86,7 +87,7 @@ class TestGenerateMain(unittest.TestCase):
|
||||
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
|
||||
|
||||
|
Reference in New Issue
Block a user