Fix extra import error and format tests with black

ajhynes7 · Mar 15, 2019 · a0eb25a · a0eb25a
1 parent e8cb5f6
commit a0eb25a
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 10 deletions.
diff --git a/skspatial/reduction.py b/skspatial/reduction.py
@@ -1,6 +1,5 @@
 """Functions related to dimension reduction."""
 
-from dpcontracts import ensure
 from numpy.linalg import matrix_rank
 
 from skspatial.transformation import mean_center

diff --git a/skspatial/transformation.py b/skspatial/transformation.py
@@ -84,12 +84,10 @@ def mean_center(points):
 
 @types(points=np.ndarray)
 @require(
-    "The points dimension cannot be greater than the desired dimension.",
-    lambda args: args.points.shape[1] <= args.dim
+    "The points dimension cannot be greater than the desired dimension.", lambda args: args.points.shape[1] <= args.dim
 )
 @ensure(
-    "The output must have the desired dimensions.",
-    lambda args, result: result.shape == (len(args.points), args.dim)
+    "The output must have the desired dimensions.", lambda args, result: result.shape == (len(args.points), args.dim)
 )
 @ensure("The output must be an ndarray.", lambda _, result: isinstance(result, np.ndarray))
 def set_dimension(points, dim):

diff --git a/tests/unit/test_fit.py b/tests/unit/test_fit.py
@@ -32,12 +32,36 @@ def test_best_fit_line(points, line_expected):
         ([[0, 0], [1, 1], [0, 2]], Plane([1 / 3, 1, 0], [0, 0, 1])),
         ([[0, 0], [0, 1], [1, 0], [1, 1]], Plane([0.5, 0.5], [0, 0, 1])),
         ([[0, 0, 0], [1, 0, 0], [0, 0, 1]], Plane([1 / 3, 0, 1 / 3], [0, 1, 0])),
-        ([[1, 0, 0], [-1, 0, 0], [1, 1, 1], [-1, 1, 1]], Plane([0, 0.5, 0.5], [0, 1, -1])),
-        ([[1, 0, 1], [1, 1, 1], [-1, 0, -1], [-1, 1, -1]], Plane([0, 0.5, 0], [1, 0, -1])),
-        ([[1, 0, 1], [1, 1, 1], [-1, 0, -1], [-1, 1, -1], [0, 0, 0]], Plane([0, 0.4, 0], [1, 0, -1])),
+        (
+            [[1, 0, 0], [-1, 0, 0], [1, 1, 1], [-1, 1, 1]],
+            Plane([0, 0.5, 0.5], [0, 1, -1]),
+        ),
+        (
+            [[1, 0, 1], [1, 1, 1], [-1, 0, -1], [-1, 1, -1]],
+            Plane([0, 0.5, 0], [1, 0, -1]),
+        ),
+        (
+            [[1, 0, 1], [1, 1, 1], [-1, 0, -1], [-1, 1, -1], [0, 0, 0]],
+            Plane([0, 0.4, 0], [1, 0, -1]),
+        ),
         # The points are not coplanar.
-        ([[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]], Plane([0.25, 0.25, 0.25], [1, 1, 1])),
-        ([[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]], Plane([0.5, 0.5, 0.5], [0, 1, 0]))
+        (
+            [[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]],
+            Plane([0.25, 0.25, 0.25], [1, 1, 1]),
+        ),
+        (
+            [
+                [0, 0, 0],
+                [0, 0, 1],
+                [0, 1, 0],
+                [0, 1, 1],
+                [1, 0, 0],
+                [1, 0, 1],
+                [1, 1, 0],
+                [1, 1, 1],
+            ],
+            Plane([0.5, 0.5, 0.5], [0, 1, 0]),
+        ),
     ],
 )
 def test_best_fit_plane(points, plane_expected):