From 79be95a991e7e3059d7f4922016c8a40028af9e6 Mon Sep 17 00:00:00 2001
From: ercdndrs <55447330+ercdndrs@users.noreply.github.com>
Date: Mon, 18 Jan 2021 22:41:30 -0500
Subject: [PATCH] v0.6-beta

---
 scripts/import_testing.py | 61 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 scripts/import_testing.py

diff --git a/scripts/import_testing.py b/scripts/import_testing.py
new file mode 100644
index 00000000..091c40a5
--- /dev/null
+++ b/scripts/import_testing.py
@@ -0,0 +1,61 @@
+import pickle
+import sys
+
+# We need to import some class definitions from the parent directory.
+from os.path import dirname, abspath, join
+base_dir = dirname(dirname(abspath(__file__)))
+sys.path.insert(1, base_dir)
+sys.path.insert(1, base_dir+'\\automaxlair')
+from automaxlair import Pokemon, Move, matchup_scoring
+
+
+def main():
+    boss_pokemon = pickle.load(open(base_dir+'/data/boss_pokemon.pickle', 'rb'))
+    rental_pokemon = pickle.load(open(base_dir+'/data/rental_pokemon.pickle', 'rb'))
+    boss_matchups = pickle.load(open(base_dir+'/data/boss_matchup_LUT.pickle', 'rb'))
+    rental_matchups = pickle.load(open(base_dir+'/data/rental_matchup_LUT.pickle', 'rb'))
+    rental_scores = pickle.load(open(base_dir+'/data/rental_pokemon_scores.pickle', 'rb'))
+
+    # Test retrieval of a rental Pokemon
+    rental_pokemon['stunfisk-galar'].print_verbose()
+    print('________________________________________')
+
+    # Test retrieval of a boss Pokemon
+    boss_pokemon['mewtwo'].print_verbose()
+    print('________________________________________')
+
+    # Test retrieval of rental Pokemon matchups
+    print(f'Matchup for Chansey against Golurk (poor): {rental_matchups["chansey"]["golurk"]}')
+    print(f'Matchup for Carkol against Butterfree (good): {rental_matchups["carkol"]["butterfree"]}')
+    print('________________________________________')
+
+    # Test retrieval of boss Pokemon matchups
+    print(f'Matchup for Jynx against Heatran (poor): {boss_matchups["jynx"]["heatran"]}')
+    print(f'Matchup for Golurk against Raikou (good): {boss_matchups["golurk"]["raikou"]}')
+    print('________________________________________')
+
+    # Test retrieval of rental Pokemon scores
+    print(f'Score for Jigglypuff (poor): {rental_scores["jigglypuff"]}')
+    print(f'Score for Doublade (good): {rental_scores["doublade"]}')
+    print('________________________________________')
+
+    # Test move selection
+    print('Wide Guard utility:')
+    matchup_scoring.print_matchup_summary(
+        rental_pokemon['pelipper'], boss_pokemon['groudon'], rental_pokemon
+    )
+    salazzle = rental_pokemon['salazzle']
+    print('Regular matchup:')
+    matchup_scoring.print_matchup_summary(
+        salazzle, boss_pokemon['kartana'], rental_pokemon
+    )
+    print('Max move scores:')
+    salazzle.dynamax = True
+    matchup_scoring.print_matchup_summary(
+        salazzle, boss_pokemon['kartana'], rental_pokemon
+    )
+
+
+
+if __name__ == '__main__':
+    main()
\ No newline at end of file