Get data dir using tests.util.get_data_dir

QuantEcon · Nov 6, 2024 · e05910a · e05910a
1 parent 8a2a49c
commit e05910a
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/quantecon/game_theory/tests/test_howson_lcp.py b/quantecon/game_theory/tests/test_howson_lcp.py
@@ -7,19 +7,26 @@
 from quantecon.game_theory.howson_lcp import polym_lcp_solver
 from quantecon.game_theory.polymatrix_game import PolymatrixGame
 
+import os
+from quantecon.tests.util import get_data_dir
+
+
+data_dir_name = "gam_files"
+data_dir = get_data_dir(data_dir_name)
+
 
 def test_polym_lcp_solver_where_solution_is_pure_NE():
-    filename = "gam_files/big_polym.gam"
-    nfg = qe_nfg_from_gam_file(filename)
+    filename = "big_polym.gam"
+    nfg = qe_nfg_from_gam_file(os.path.join(data_dir, filename))
     polym = PolymatrixGame.from_nf(nfg)
     ne = polym_lcp_solver(polym)
     worked = nfg.is_nash(ne)
     assert_(worked)
 
 
 def test_polym_lcp_solver_where_lcp_solver_must_backtrack():
-    filename = "gam_files/triggers_back_case.gam"
-    nfg = qe_nfg_from_gam_file(filename)
+    filename = "triggers_back_case.gam"
+    nfg = qe_nfg_from_gam_file(os.path.join(data_dir, filename))
     polym = PolymatrixGame.from_nf(nfg)
     ne = polym_lcp_solver(polym)
     worked = nfg.is_nash(ne)