fix formatting and tests

Author: dbochkov-flexcompute

tests/test_components/test_grid_spec.py

@@ -542,27 +542,60 @@ def test_uniform_grid_dl_validation(dl, expect_exception):
             run_time=1e-12,
         )
 
-        
-def test_dl_min_from_smallest_feature():
 
+def test_dl_min_from_smallest_feature():
     structure = td.Structure(
-        geometry=td.PolySlab(vertices=[[0, 0], [2, 0], [2, 1], [1, 1], [1, 1.1], [2, 1.1], [2, 2], [1, 2], [1, 2.2], [0.7, 2.2], [0.7, 2], [0, 2]], slab_bounds=[-1, 1], axis=2),
+        geometry=td.PolySlab(
+            vertices=[
+                [0, 0],
+                [2, 0],
+                [2, 1],
+                [1, 1],
+                [1, 1.1],
+                [2, 1.1],
+                [2, 2],
+                [1, 2],
+                [1, 2.2],
+                [0.7, 2.2],
+                [0.7, 2],
+                [0, 2],
+            ],
+            slab_bounds=[-1, 1],
+            axis=2,
+        ),
         medium=td.PECMedium(),
     )
 
     # check expected dl_min
-    layer_spec = td.LayerRefinementSpec(axis=2, size=(td.inf, td.inf, 2), corner_finder=td.CornerFinderSpec(dl_min_spec=td.SmallestFeatureSpec(convex_resolution=10, concave_resolution=0, mixed_resolution=0)))
+    layer_spec = td.LayerRefinementSpec(
+        axis=2,
+        size=(td.inf, td.inf, 2),
+        corner_finder=td.CornerFinderSpec(
+            convex_resolution=10, concave_resolution=None, mixed_resolution=None
+        ),
+    )
     dl_min = layer_spec._dl_min_from_smallest_feature([structure])
-    assert(np.allclose(0.3 / 10, dl_min))
+    assert np.allclose(0.3 / 10, dl_min)
 
-    layer_spec = td.LayerRefinementSpec(axis=2, size=(td.inf, td.inf, 2), corner_finder=td.CornerFinderSpec(dl_min_spec=td.SmallestFeatureSpec(convex_resolution=0, concave_resolution=0, mixed_resolution=10)))
+    layer_spec = td.LayerRefinementSpec(
+        axis=2,
+        size=(td.inf, td.inf, 2),
+        corner_finder=td.CornerFinderSpec(
+            convex_resolution=None, concave_resolution=None, mixed_resolution=10
+        ),
+    )
     dl_min = layer_spec._dl_min_from_smallest_feature([structure])
-    assert(np.allclose(0.2 / 10, dl_min))
+    assert np.allclose(0.2 / 10, dl_min)
 
-    layer_spec = td.LayerRefinementSpec(axis=2, size=(td.inf, td.inf, 2), corner_finder=td.CornerFinderSpec(dl_min_spec=td.SmallestFeatureSpec(convex_resolution=0, concave_resolution=10, mixed_resolution=0)))
+    layer_spec = td.LayerRefinementSpec(
+        axis=2,
+        size=(td.inf, td.inf, 2),
+        corner_finder=td.CornerFinderSpec(
+            convex_resolution=None, concave_resolution=10, mixed_resolution=None
+        ),
+    )
     dl_min = layer_spec._dl_min_from_smallest_feature([structure])
-    assert(np.allclose(0.1 / 10, dl_min))
-
+    assert np.allclose(0.1 / 10, dl_min)
 
     # check grid is generated succesfully
     sim = td.Simulation(
@@ -573,4 +606,3 @@ def test_dl_min_from_smallest_feature():
     )
 
     _ = sim.grid
-