Formatting/cleanups.

exact_cover/wrapper.py

@@ -41,6 +41,7 @@ def truncate(row):
                 return truncate(row[:-1])
             else:
                 return row
+
     return set([tuple(truncate(row)) for row in a])
 
 
@@ -58,4 +59,4 @@ def get_all_solutions(matrix, max_count=None):
     result = raw_get_all_solutions(transformed, count)
     if result.size == 0:
         raise NoSolution("No solutions found by the C code.")
-    return _solutions_array_to_set(result) 
+    return _solutions_array_to_set(result)

tests/problems.py

@@ -172,5 +172,5 @@ def bruteforce_problem3():
             {5, 9, 10},
             {6, 7, 8},
             {9, 10, 11},
-        ]
+        ],
     )

tests/test_all_solutions.py

@@ -12,15 +12,18 @@ def normalize(solution):
     """
     return frozenset(frozenset(row) for row in solution)
 
+
 def make_test_from_problem(problem):
     """
     generate a test function for a problem dict
     """
+
     def test():
         # xxx not yet available
         result = normalize(get_all_solutions(problem["data"]))
         expected = normalize(problem["all_solutions"])
         assert result == expected
+
     return test
 
 
@@ -31,9 +34,8 @@ def test():
 # - and not be a private symbol (start with '_')
 
 for problem_name in dir(problems):
-    if ('_problem' not in problem_name
-        or problem_name.startswith('_')):
+    if "_problem" not in problem_name or problem_name.startswith("_"):
         continue
-    test_name = 'test_' + problem_name
+    test_name = "test_" + problem_name
     problem_function = getattr(problems, problem_name)
     globals()[test_name] = make_test_from_problem(problem_function())

tests/test_count_solutions.py

@@ -5,7 +5,6 @@
 from hypothesis.strategies import sampled_from
 
 from exact_cover import get_solution_count
-from exact_cover.error import NoSolution
 from exact_cover.io import DTYPE_FOR_ARRAY
 from .helpers.polyomino_data import polyomino_problem
 from .test_exact_cover_problems import large_problems_without_solution

tests/test_get_all_solutions.py

@@ -1,14 +1,7 @@
 import numpy as np
-import pytest
-
-from hypothesis import given
-from hypothesis.strategies import sampled_from
 
 from exact_cover import get_all_solutions
-from exact_cover.error import NoSolution
 from exact_cover.io import DTYPE_FOR_ARRAY
-from .helpers.polyomino_data import polyomino_problem
-from .test_exact_cover_problems import large_problems_without_solution
 
 
 def test_exact_cover():