about summary refs log tree commit diff
diff options
context:
space:
mode:
-rw-r--r--tests/test_pyhegp.py2
-rw-r--r--tests/test_serialization.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_pyhegp.py b/tests/test_pyhegp.py
index a5a6697..a7c78a9 100644
--- a/tests/test_pyhegp.py
+++ b/tests/test_pyhegp.py
@@ -32,7 +32,7 @@ from pyhegp.pyhegp import Stats, main, hegp_encrypt, hegp_decrypt, random_key, p
 from pyhegp.serialization import Summary, read_summary, read_genotype, is_genotype_metadata_column, is_phenotype_metadata_column
 from pyhegp.utils import negate
 
-from helpers.strategies import *
+from helpers.strategies import genotype_frames, phenotype_frames, keys
 
 @given(st.lists(st.lists(arrays("float64",
                                 st.shared(array_shapes(min_dims=1, max_dims=1),
diff --git a/tests/test_serialization.py b/tests/test_serialization.py
index 32cb58f..690e968 100644
--- a/tests/test_serialization.py
+++ b/tests/test_serialization.py
@@ -25,7 +25,7 @@ from pytest import approx
 
 from pyhegp.serialization import read_summary, write_summary, read_summary_headers, read_genotype, write_genotype, read_phenotype, write_phenotype, read_key, write_key
 
-from helpers.strategies import *
+from helpers.strategies import summaries, genotype_frames, phenotype_frames, keys
 
 @given(summaries())
 def test_read_write_summary_are_inverses(summary):