Optimize db ops

.gitignore

@@ -51,6 +51,10 @@ coverage.xml
 .hypothesis/
 .pytest_cache/
 cover/
+tests/sensor/sweeps/outputs
+doc/html
+uv.lock
+pixi.lock
 
 # Jupyter Notebook
 .ipynb_checkpoints

pixi.toml

@@ -25,6 +25,7 @@ array-api-compat = "*"
 cython = "*"
 dask = "*"
 numba = ">=0.56"
+numcodecs = "*"
 numexpr = ">=2.8"
 numpy = ">=1.21"
 netcdf4 = "*"

pyproject.toml

@@ -60,7 +60,8 @@ gpu = [
 test = [
     "pytest",
     "nptdms",
-    "pytest-order"
+    "pytest-order",
+    "hypothesis"
 ]
 
 [project.scripts]
@@ -113,3 +114,16 @@ norecursedirs = [
 call-top-callable = "ignore"
 invalid-type-form = "ignore"
 
+[tool.tox]
+# Forces tox to auto-provision the tox-uv plugin if missing
+requires = ["tox-uv"]
+env_list = ["py39", "py314"]
+
+[tool.tox.env_run_base]
+# Dependencies specified JSON-style for TOML compatibility
+package = "editable"
+deps = ["pytest"]
+extras = ["test"]
+commands = [
+    ["pytest"]
+]

src/striqt/analysis/measurements/_cellular_resource_power_histogram.py

@@ -292,7 +292,7 @@ def cellular_resource_power_histogram(iq: 'Array', capture: specs.Capture, **kwa
     if time_bin_averaging is not None:
         masked_spgs = sw.binned_mean(masked_spgs, time_bin_averaging, axis=2, fft=False)
 
-    masked_spgs = sw.powtodB(masked_spgs, out=masked_spgs)
+    masked_spgs = sw.powtodB(masked_spgs, overwrite_x=True)
     bin_edges = _channel_power_histogram.make_power_histogram_bin_edges(
         power_low=spec.power_low,
         power_high=spec.power_high,

src/striqt/waveform/lib/arrays.py

@@ -578,22 +578,7 @@ def array_stream(obj: Array, null=False, non_blocking=False, ptds=False):
 
 
 def array_namespace(a, use_compat=False) -> ModuleType:
-    try:
-        return array_api_compat.array_namespace(a, use_compat=use_compat)
-    except TypeError:
-        pass
-
-    try:
-        import mlx.core as mx  # type: ignore
-
-        if isinstance(a, mx.array):
-            return mx
-        else:
-            raise TypeError
-    except (ImportError, TypeError):
-        pass
-
-    raise TypeError('unrecognized object type')
+    return array_api_compat.array_namespace(a, use_compat=use_compat)
 
 
 def convert_np_to_xp(func: _TC) -> _TC:

src/striqt/waveform/lib/fourier.py

@@ -319,7 +319,7 @@ def fftfreq(
         values = [fres * n for n in span]
     else:
         values = [fres * (n + 1) for n in span]
-    return np.array(values, dtype=dtype)
+    return xp.array(values, dtype=dtype)
 
 
 def time_fftshift(x, scale: Array | float | None = None, overwrite_x=False, axis=0):
@@ -589,10 +589,10 @@ def stft(
         nfft=nfft,
         time_size=y.shape[axis],
         overlap_frac=noverlap / nfft,
-        xp=np,
+        xp=xp,
     )
 
-    return typing.cast(tuple[_AT, _AT, _AT], (freqs, times, y))
+    return (freqs, times, y)  # type: ignore
 
 
 def istft(