maint[tests]: reduce runtime of most expensive tests

[yaugenst-flex]

Shaves a good amount of seconds off from the slowest running tests. Introduces a fixture that allows disabling most of the td.Simulation validation. Some of these are still slower than they should be probably but it's a start.

Before:

75.52s call     tests/test_plugins/test_design.py::test_sweep[sweep_method0]
40.89s call     tests/test_components/test_autograd.py::test_autograd_async[<ALL>-diff]
34.13s call     tests/test_components/test_autograd.py::test_autograd_async[<ALL>-field_vol]
33.85s call     tests/test_components/test_simulation.py::test_num_lumped_elements
31.11s call     tests/test_components/test_autograd.py::test_autograd_async[<ALL>-mode]
24.55s call     tests/test_components/test_autograd.py::test_autograd_async[<ALL>-field_point]
22.17s call     tests/test_plugins/test_adjoint.py::test_adjoint_pipeline[True]
20.26s call     tests/test_components/test_scene.py::test_num_mediums
19.62s call     tests/test_components/test_IO.py::test_validation_speed
19.13s call     tests/test_components/test_autograd.py::test_autograd_async_server[<ALL>-diff]

After:

42.02s call     tests/test_components/test_simulation.py::test_num_lumped_elements
24.76s call     tests/test_plugins/test_adjoint.py::TestJaxComplexPolySlab::test_vertices_grads[-0.02-10-3-base_vertices1]
24.61s call     tests/test_components/test_autograd.py::test_autograd_async_server[<ALL>-diff]
23.64s call     tests/test_plugins/test_adjoint.py::test_adjoint_pipeline[True]
23.14s call     tests/test_components/test_autograd.py::test_autograd_async_some_zero_grad[<ALL>-field_vol]
22.75s call     tests/test_components/test_autograd.py::test_autograd_async[<ALL>-diff]
22.29s call     tests/test_components/test_eme.py::test_eme_sim_data
21.59s call     tests/test_components/test_autograd.py::test_autograd_async_some_zero_grad[<ALL>-mode]
21.50s call     tests/test_components/test_scene.py::test_num_mediums
21.35s call     tests/test_components/test_autograd.py::test_autograd_async_server[<ALL>-field_vol]

Note: Not sure where the 42s spent on test_num_lumped_elements comes from, might have gotten unlucky on an efficiency core.

[momchil-flex]

Interesting.

• Why does validation take so long in the test where it does?
• I thought that there was no good "construct" method that works right off the bat? Like I thought that it doesn't initialize embedded models, you get something like e.g. Simulation looks like the proper object but an embedded e.g. Monitor would look like a dict rather than a Monitor.

[yaugenst-flex]

• Why does validation take so long in the test where it does?

The test_design.py::test_sweep was just assembling ~700 simulations, and validating those amounted to ~20% of the total time. I also reduced the size of the sweep, that was probably a bigger speedup.

• I thought that there was no good "construct" method that works right off the bat? Like I thought that it doesn't initialize embedded models, you get something like e.g. Simulation looks like the proper object but an embedded e.g. Monitor would look like a dict rather than a Monitor.

At least in the tests we are generally passing already-validated submodels, and in that case construct() works fine. It's just a problem if you're passing dicts around I think.

Anyways, this PR is not done I only undrafted it to check out the CI. I'll keep this one open for a while and maybe we can speed up the tests a bit more.