First attempts

TorchDSP · Sep 5, 2023 · 594b1a5 · 594b1a5
1 parent 157626d
commit 594b1a5
Show file tree

Hide file tree

Showing 6 changed files with 261 additions and 140 deletions.
diff --git a/tests/test_transforms_benchmark.py b/tests/test_transforms_benchmark.py
@@ -25,60 +25,60 @@ def generate_data():
 
 
 transforms_list = [
-    (
-        "random_resample_up",
-        RandomResample(1.5, num_iq_samples=128, keep_samples=False),
-        RandomResample(1.5, num_iq_samples=4096, keep_samples=False),
-    ),
-    (
-        "random_resample_down",
-        RandomResample(0.75, num_iq_samples=128, keep_samples=False),
-        RandomResample(0.75, num_iq_samples=4096, keep_samples=False),
-    ),
+    # (
+    #     "random_resample_up",
+    #     RandomResample(1.5, num_iq_samples=128, keep_samples=False),
+    #     RandomResample(1.5, num_iq_samples=4096, keep_samples=False),
+    # ),
+    # (
+    #     "random_resample_down",
+    #     RandomResample(0.75, num_iq_samples=128, keep_samples=False),
+    #     RandomResample(0.75, num_iq_samples=4096, keep_samples=False),
+    # ),
     ("add_noise", AddNoise(-10), AddNoise(-10)),
-    ("time_varying_noise", TimeVaryingNoise(-30, -10), TimeVaryingNoise(-30, -10)),
-    (
-        "rayleigh_fading",
-        RayleighFadingChannel(0.05, (1.0, 0.5, 0.1)),
-        RayleighFadingChannel(0.05, (1.0, 0.5, 0.1)),
-    ),
-    ("phase_shift", RandomPhaseShift(0.5), RandomPhaseShift(0.5)),
-    ("time_shift", RandomTimeShift(-100.5), RandomTimeShift(-2.5)),
-    (
-        "time_crop",
-        TimeCrop("random", length=128),
-        TimeCrop("random", length=4096),
-    ),
-    ("time_reversal", TimeReversal(False), TimeReversal(False)),
-    ("frequency_shift", RandomFrequencyShift(-0.25), RandomFrequencyShift(-0.25)),
-    (
-        "delayed_frequency_shift",
-        RandomDelayedFrequencyShift(0.2, 0.25),
-        RandomDelayedFrequencyShift(0.2, 0.25),
-    ),
-    (
-        "oscillator_drift",
-        LocalOscillatorDrift(0.01, 0.001),
-        LocalOscillatorDrift(0.01, 0.001),
-    ),
-    ("gain_drift", GainDrift(0.01, 0.001, 0.1), GainDrift(0.01, 0.001, 0.1)),
-    (
-        "iq_imbalance",
-        IQImbalance(3, np.pi / 180, 0.05),
-        IQImbalance(3, np.pi / 180, 0.05),
-    ),
-    ("roll_off", RollOff(0.05, 0.98), RollOff(0.05, 0.98)),
-    ("add_slope", AddSlope(), AddSlope()),
-    ("spectral_inversion", SpectralInversion(), SpectralInversion()),
-    ("channel_swap", ChannelSwap(), ChannelSwap()),
-    ("magnitude_rescale", RandomMagRescale(0.5, 3), RandomMagRescale(0.5, 3)),
-    (
-        "drop_samples",
-        RandomDropSamples(0.3, 50, ["zero"]),
-        RandomDropSamples(0.3, 50, ["zero"]),
-    ),
-    ("quantize", Quantize(32, ["floor"]), Quantize(32, ["floor"])),
-    ("clip", Clip(0.85), Clip(0.85)),
+    # ("time_varying_noise", TimeVaryingNoise(-30, -10), TimeVaryingNoise(-30, -10)),
+    # (
+    #     "rayleigh_fading",
+    #     RayleighFadingChannel(0.05, (1.0, 0.5, 0.1)),
+    #     RayleighFadingChannel(0.05, (1.0, 0.5, 0.1)),
+    # ),
+    # ("phase_shift", RandomPhaseShift(0.5), RandomPhaseShift(0.5)),
+    # ("time_shift", RandomTimeShift(-100.5), RandomTimeShift(-2.5)),
+    # (
+    #     "time_crop",
+    #     TimeCrop("random", length=128),
+    #     TimeCrop("random", length=4096),
+    # ),
+    # ("time_reversal", TimeReversal(False), TimeReversal(False)),
+    # ("frequency_shift", RandomFrequencyShift(-0.25), RandomFrequencyShift(-0.25)),
+    # (
+    #     "delayed_frequency_shift",
+    #     RandomDelayedFrequencyShift(0.2, 0.25),
+    #     RandomDelayedFrequencyShift(0.2, 0.25),
+    # ),
+    # (
+    #     "oscillator_drift",
+    #     LocalOscillatorDrift(0.01, 0.001),
+    #     LocalOscillatorDrift(0.01, 0.001),
+    # ),
+    # ("gain_drift", GainDrift(0.01, 0.001, 0.1), GainDrift(0.01, 0.001, 0.1)),
+    # (
+    #     "iq_imbalance",
+    #     IQImbalance(3, np.pi / 180, 0.05),
+    #     IQImbalance(3, np.pi / 180, 0.05),
+    # ),
+    # ("roll_off", RollOff(0.05, 0.98), RollOff(0.05, 0.98)),
+    # ("add_slope", AddSlope(), AddSlope()),
+    # ("spectral_inversion", SpectralInversion(), SpectralInversion()),
+    # ("channel_swap", ChannelSwap(), ChannelSwap()),
+    # ("magnitude_rescale", RandomMagRescale(0.5, 3), RandomMagRescale(0.5, 3)),
+    # (
+    #     "drop_samples",
+    #     RandomDropSamples(0.01, 50, ["zero"]),
+    #     RandomDropSamples(0.01, 50, ["zero"]),
+    # ),
+    # ("quantize", Quantize(32, ["floor"]), Quantize(32, ["floor"])),
+    # ("clip", Clip(0.85), Clip(0.85)),
 ]
 
 

diff --git a/tests/test_transforms_figures.py b/tests/test_transforms_figures.py
@@ -101,7 +101,12 @@ def generate_data(modulation_name):
 
 
 @pytest.mark.parametrize(
-    "transform, modulation_name", itertools.product(transforms_list, modulations)
+    "transform, modulation_name",
+    itertools.product(transforms_list, modulations),
+    ids=[
+        "{}-{}".format(x[0], y)
+        for x, y in itertools.product(transforms_list, modulations)
+    ],
 )
 def test_transform_figures(transform, modulation_name):
     short_data, long_data = generate_data(modulation_name)