reformatted using black . --line-length 100

Author: jewettaijfc

tests/test_components/test_IO.py

@@ -73,19 +73,19 @@ def test_simulation_load_export(split_string, tmp_path):
     SIM.to_hdf5(path_hdf5)
     SIM2 = td.Simulation.from_file(path)
     SIM_HDF5 = td.Simulation.from_hdf5(path_hdf5)
-    assert set_datasets_to_none(SIM)._json_string == SIM2._json_string, (
-        "original and loaded simulations are not the same"
-    )
+    assert (
+        set_datasets_to_none(SIM)._json_string == SIM2._json_string
+    ), "original and loaded simulations are not the same"
     assert SIM == SIM_HDF5, "original and loaded from hdf5 simulations are not the same"
 
 
 def test_simulation_load_export_yaml(tmp_path):
     path = str(tmp_path / "simulation.yaml")
     SIM.to_file(path)
     SIM2 = td.Simulation.from_file(path)
-    assert set_datasets_to_none(SIM)._json_string == SIM2._json_string, (
-        "original and loaded simulations are not the same"
-    )
+    assert (
+        set_datasets_to_none(SIM)._json_string == SIM2._json_string
+    ), "original and loaded simulations are not the same"
 
 
 def test_component_load_export(tmp_path):

tests/test_components/test_autograd.py

@@ -1247,18 +1247,18 @@ def test_adjoint_src_width():
     adj_srcs_fwidth = td.SimulationData._adjoint_src_width_single(adj_srcs)
 
     for src in adj_srcs_fwidth:
-        assert np.isclose((src.source_time.freq0 - f0) / f0, 0.0), (
-            "f0 of adjoint source should be centered on original f0"
-        )
+        assert np.isclose(
+            (src.source_time.freq0 - f0) / f0, 0.0
+        ), "f0 of adjoint source should be centered on original f0"
 
         check_fwidth = (
             src.source_time.freq0
             - td.components.data.sim_data.NUM_ADJOINT_FWIDTH_TO_ZERO * src.source_time.fwidth
         ) / src.source_time.freq0
 
-        assert np.isclose(check_fwidth, 0.0) or (check_fwidth > 0.0), (
-            "fwidth of adjoint source should decay sufficiently before f=0"
-        )
+        assert np.isclose(check_fwidth, 0.0) or (
+            check_fwidth > 0.0
+        ), "fwidth of adjoint source should decay sufficiently before f=0"
 
 
 def test_broadband_adjoint_src_width():
@@ -1305,12 +1305,12 @@ def test_broadband_adjoint_src_width():
             f0_expected - np.min(f0_adj_all)
         ) / td.components.data.sim_data.NUM_ADJOINT_FWIDTH_TO_FMIN
 
-        assert np.isclose((f0_expected - broadband_f0) / f0_expected, 0.0), (
-            "Expected freq0 not matching for broadband source"
-        )
-        assert np.isclose((fwidth_expected - broadband_fwidth) / fwidth_expected, 0.0), (
-            "Expected fwidth not matching for broadband source"
-        )
+        assert np.isclose(
+            (f0_expected - broadband_f0) / f0_expected, 0.0
+        ), "Expected freq0 not matching for broadband source"
+        assert np.isclose(
+            (fwidth_expected - broadband_fwidth) / fwidth_expected, 0.0
+        ), "Expected fwidth not matching for broadband source"
 
     # Test the case where we need a wider pulse to cover all the adjoint frequencies than we would otherwise choose for
     # each individual adjoint source
@@ -1333,12 +1333,12 @@ def test_broadband_adjoint_src_width():
         f0_expected - np.min(f0_broadband)
     ) / td.components.data.sim_data.NUM_ADJOINT_FWIDTH_TO_FMIN
 
-    assert np.isclose((f0_expected - broadband_f0) / f0_expected, 0.0), (
-        "Expected freq0 not matching for broadband source"
-    )
-    assert np.isclose((fwidth_expected - broadband_fwidth) / fwidth_expected, 0.0), (
-        "Expected fwidth not matching for broadband source"
-    )
+    assert np.isclose(
+        (f0_expected - broadband_f0) / f0_expected, 0.0
+    ), "Expected freq0 not matching for broadband source"
+    assert np.isclose(
+        (fwidth_expected - broadband_fwidth) / fwidth_expected, 0.0
+    ), "Expected fwidth not matching for broadband source"
 
     # Test the case where we have a narrow set of frequencies for the adjoint sources and so we can
     # choose a wider overall source than is needed for covering those frequencies. This larger pulse width
@@ -1360,12 +1360,12 @@ def test_broadband_adjoint_src_width():
     f0_expected = 0.5 * (np.max(f0_broadband) + np.min(f0_broadband))
     fwidth_expected = f0_expected / td.components.data.sim_data.NUM_ADJOINT_FWIDTH_TO_ZERO
 
-    assert np.isclose((f0_expected - broadband_f0) / f0_expected, 0.0), (
-        "Expected freq0 not matching for broadband source"
-    )
-    assert np.isclose((fwidth_expected - broadband_fwidth) / fwidth_expected, 0.0), (
-        "Expected fwidth not matching for broadband source"
-    )
+    assert np.isclose(
+        (f0_expected - broadband_f0) / f0_expected, 0.0
+    ), "Expected freq0 not matching for broadband source"
+    assert np.isclose(
+        (fwidth_expected - broadband_fwidth) / fwidth_expected, 0.0
+    ), "Expected fwidth not matching for broadband source"
 
 
 @pytest.mark.parametrize("colocate", [True, False])

tests/test_components/test_autograd_mode_polyslab_numerical.py

@@ -332,7 +332,9 @@ def test_finite_difference_mode_data_polyslab(
         "geometry_size_wvl",
         "polyslab_permittivity",
         "test_number",
-    )(mode_data_test_parameters)
+    )(
+        mode_data_test_parameters
+    )
 
     adj_freq = td.C_0 / adj_wvl_um
 
@@ -372,11 +374,7 @@ def eval_fn(sim_data):
     polyslab_height_um = POLYSLAB_HEIGHT_WVL * adj_wvl_um
 
     objective = create_objective_function(
-        lambda mesh_wvl_um=mesh_wvl_um,
-        adj_wvl_um=adj_wvl_um,
-        geometry_size_wvl=geometry_size_wvl,
-        polyslab_permittivity=polyslab_permittivity,
-        box_for_override=box_for_override: make_base_sim(
+        lambda mesh_wvl_um=mesh_wvl_um, adj_wvl_um=adj_wvl_um, geometry_size_wvl=geometry_size_wvl, polyslab_permittivity=polyslab_permittivity, box_for_override=box_for_override: make_base_sim(
             mesh_wvl_um=mesh_wvl_um,
             adj_wvl_um=adj_wvl_um,
             geometry_size_wvl=geometry_size_wvl,

tests/test_components/test_autograd_numerical.py

@@ -256,7 +256,9 @@ def test_finite_difference_field_data(field_data_test_parameters, rng, tmp_path,
         "eval_fn",
         "eval_fn_name",
         "test_number",
-    )(field_data_test_parameters)
+    )(
+        field_data_test_parameters
+    )
 
     dim_um = mesh_wvl_um
     dim_um = mesh_wvl_um
@@ -279,11 +281,7 @@ def test_finite_difference_field_data(field_data_test_parameters, rng, tmp_path,
 
     objective = create_objective_function(
         block,
-        lambda mesh_wvl_um=mesh_wvl_um,
-        adj_wvl_um=adj_wvl_um,
-        monitor_size_wvl=monitor_size_wvl,
-        box_for_override=box_for_override,
-        monitor_bg_index=monitor_bg_index: make_base_sim(
+        lambda mesh_wvl_um=mesh_wvl_um, adj_wvl_um=adj_wvl_um, monitor_size_wvl=monitor_size_wvl, box_for_override=box_for_override, monitor_bg_index=monitor_bg_index: make_base_sim(
             mesh_wvl_um=mesh_wvl_um,
             adj_wvl_um=adj_wvl_um,
             monitor_size_wvl=monitor_size_wvl,

tests/test_components/test_eme.py

@@ -282,6 +282,7 @@ def test_eme_monitor():
         center=(1, 2, 3), size=(2, 2, 2), freqs=[300e12], num_modes=2, name="eme_coeffs"
     )
 
+
 @pytest.mark.parametrize("transpose", [True, False])
 def test_eme_simulation(transpose):
     sim = make_eme_sim()

tests/test_components/test_geometry.py

@@ -90,6 +90,7 @@ def test_plot(component, transpose):
     _ = component.plot(z=0, ax=AX, transpose=transpose)
     plt.close()
 
+
 @pytest.mark.parametrize("transpose", [True, False])
 def test_plot_with_units(transpose):
     _ = BOX.plot(z=0, ax=AX, plot_length_units="nm", transpose=transpose)
@@ -329,15 +330,15 @@ def test_polyslab_inf_to_finite_bounds(axis):
         vertices=[[0, 0], [2.5, 1], [2, 3], [0.5, 4], [-1.5, 2.5]],
     )
 
-    assert ps_low_inf.finite_length_axis == (LARGE_NUMBER + axis_bound), (
-        "Unexpected finite length for polyslab axis with -inf bound"
-    )
-    assert ps_high_inf.finite_length_axis == (LARGE_NUMBER + axis_bound), (
-        "Unexpected finite length for polyslab axis with inf bound"
-    )
-    assert ps_inf.finite_length_axis == 2 * LARGE_NUMBER, (
-        "Unexpected finite length for polyslab axis with two inf bounds"
-    )
+    assert ps_low_inf.finite_length_axis == (
+        LARGE_NUMBER + axis_bound
+    ), "Unexpected finite length for polyslab axis with -inf bound"
+    assert ps_high_inf.finite_length_axis == (
+        LARGE_NUMBER + axis_bound
+    ), "Unexpected finite length for polyslab axis with inf bound"
+    assert (
+        ps_inf.finite_length_axis == 2 * LARGE_NUMBER
+    ), "Unexpected finite length for polyslab axis with two inf bounds"
 
 
 def test_validate_polyslab_vertices_valid():

tests/test_components/test_heat_charge.py

@@ -868,14 +868,14 @@ def test_heat_charge_simulation(transpose, simulation_data):
     assert cond_sim is not None, "Conduction simulation should be created successfully."
 
     voltage_capacitance_sim = voltage_capacitance_sim_data.simulation
-    assert voltage_capacitance_sim is not None, (
-        "Voltage-Capacitance simulation should be created successfully."
-    )
+    assert (
+        voltage_capacitance_sim is not None
+    ), "Voltage-Capacitance simulation should be created successfully."
 
     current_voltage_sim = current_voltage_simulation_data.simulation
-    assert current_voltage_sim is not None, (
-        "Current-Voltage simulation should be created successfully."
-    )
+    assert (
+        current_voltage_sim is not None
+    ), "Current-Voltage simulation should be created successfully."
 
     _ = heat_sim.plot_heat_conductivity(x=0, transpose=transpose)
     plt.close()

tests/test_components/test_meshgenerate.py

@@ -818,9 +818,9 @@ def test_override_are_box():
             dl=[1, 2, 3],
         )
 
-    assert isinstance(override_not_box.geometry, td.Box), (
-        "Sphere override structure was not converted to Box"
-    )
+    assert isinstance(
+        override_not_box.geometry, td.Box
+    ), "Sphere override structure was not converted to Box"
 
 
 def test_override_unshadowed():

tests/test_components/test_simulation.py

@@ -2901,9 +2901,11 @@ def test_sim_subsection(unstructured, nz):
         boundary_spec=td.BoundarySpec.all_sides(td.Periodic()),
         # Set theta to 'pi/2' for 2D simulation in the x-y plane
         monitors=[
-            mnt.updated_copy(theta=np.pi / 2)
-            if isinstance(mnt, td.FieldProjectionAngleMonitor)
-            else mnt
+            (
+                mnt.updated_copy(theta=np.pi / 2)
+                if isinstance(mnt, td.FieldProjectionAngleMonitor)
+                else mnt
+            )
             for mnt in subsection_monitors
             if not isinstance(
                 mnt, (td.FieldProjectionCartesianMonitor, td.FieldProjectionKSpaceMonitor)

tests/test_material_library/test_material_library.py

@@ -71,20 +71,20 @@ def test_medium_repr():
     repr_noname_medium = test_media[0].__repr__()
     str_noname_medium_dict = str(noname_medium_in_dict)
 
-    assert "type='Medium' permittivity=2.25 conductivity=0.0" in str_noname_medium, (
-        "Expected medium information in string"
-    )
-    assert "Medium(attrs={}, name=None, frequency_range=None" in repr_noname_medium, (
-        "Expcted medium information in repr"
-    )
+    assert (
+        "type='Medium' permittivity=2.25 conductivity=0.0" in str_noname_medium
+    ), "Expected medium information in string"
+    assert (
+        "Medium(attrs={}, name=None, frequency_range=None" in repr_noname_medium
+    ), "Expcted medium information in repr"
     assert repr_noname_medium in str_noname_medium_dict, "Expected repr in dictionary string"
 
     for medium in test_media:
         repr_str = medium.__repr__()
 
-    assert test_media[1].__repr__() == material_name, (
-        "Expected repr to return just the material name."
-    )
+    assert (
+        test_media[1].__repr__() == material_name
+    ), "Expected repr to return just the material name."
 
 
 def test_variant_str():
@@ -97,16 +97,16 @@ def test_variant_str():
 
     printed_SiO2_Palik_lossless = str(td.material_library["SiO2"].variants["Palik_Lossless"])
 
-    assert "eps_inf: 1.5385442336875639" in printed_SiO2_Palik_lossless, (
-        "Expected eps_inf in SiO2 printed string"
-    )
+    assert (
+        "eps_inf: 1.5385442336875639" in printed_SiO2_Palik_lossless
+    ), "Expected eps_inf in SiO2 printed string"
     assert "poles: 2" in printed_SiO2_Palik_lossless, "Expected 1 pole in SiO2 printed string"
 
     printed_SiO2_Palik_lossy = str(td.material_library["SiO2"].variants["Palik_Lossy"])
 
-    assert "eps_inf: 2.1560362571240765" in printed_SiO2_Palik_lossy, (
-        "Expected eps_inf in SiO2 printed string"
-    )
+    assert (
+        "eps_inf: 2.1560362571240765" in printed_SiO2_Palik_lossy
+    ), "Expected eps_inf in SiO2 printed string"
     assert "poles: 5" in printed_SiO2_Palik_lossy, "Expected 1 pole in SiO2 printed string"
 
 
@@ -115,9 +115,9 @@ def test_material_str():
 
     printed_Ag = str(td.material_library["Ag"])
 
-    assert "Default Variant: Rakic1998BB" in printed_Ag, (
-        "Expected default variant in printed string"
-    )
+    assert (
+        "Default Variant: Rakic1998BB" in printed_Ag
+    ), "Expected default variant in printed string"
     assert "RakicLorentzDrude1998" in printed_Ag, "Expected variant in printed string"
 
     printed_Au = str(td.material_library["Au"])