Reformatted files

tests/test_preprocess.py

@@ -210,7 +210,9 @@ def test_get_table_dicts(nodes, ego, sampler):
 
     if table._type == "coords" and algn == "Nose" and polar is False and speed == 0:
 
-        assert type(table.plot_heatmaps(bodyparts=["Spine_1"])) == matplotlib.figure.Figure
+        assert (
+            type(table.plot_heatmaps(bodyparts=["Spine_1"])) == matplotlib.figure.Figure
+        )
 
         align = sampler.draw(
             st.one_of(st.just(False), st.just("all"), st.just("center"))

tests/test_visuals.py

@@ -36,11 +36,7 @@ def test_plot_heatmap(bparts):
     assert (
         type(
             deepof.visuals.plot_heatmap(
-                prun["test"],
-                bparts,
-                tuple([-100, 100]),
-                tuple([-100, 100]),
-                dpi=200,
+                prun["test"], bparts, tuple([-100, 100]), tuple([-100, 100]), dpi=200,
             )
         )
         == matplotlib.figure.Figure