Update type chart tests

hsahovic · Oct 31, 2019 · 5322815 · 5322815
1 parent 3319ab1
commit 5322815
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/test_utils.py → test/test_data.py b/test/test_utils.py → test/test_data.py
@@ -1,11 +1,12 @@
 # -*- coding: utf-8 -*-
 
-from poke_env.utils import compute_type_chart, TYPE_CHART
+from poke_env.data import TYPE_CHART, TYPE_CHART_PATH
+from poke_env.utils import compute_type_chart
 from poke_env.environment.pokemon_type import PokemonType
 
 
 def test_type_chart_corresponds_to_compute_type_chart():
-    assert compute_type_chart() == TYPE_CHART
+    assert compute_type_chart(TYPE_CHART_PATH) == TYPE_CHART
 
 
 def test_types_correspond_to_type_chart_entries():