Continuning cleanup and vscode integration

misko · Dec 29, 2023 · 64475c7 · 64475c7
1 parent 9c29bce
commit 64475c7
Show file tree

Hide file tree

Showing 12 changed files with 79 additions and 269 deletions.
diff --git a/.env b/.env
diff --git a/.ipynb_checkpoints/polygon_testing-checkpoint.ipynb b/.ipynb_checkpoints/polygon_testing-checkpoint.ipynb
@@ -0,0 +1,6 @@
+{
+ "cells": [],
+ "metadata": {},
+ "nbformat": 4,
+ "nbformat_minor": 5
+}
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -0,0 +1,15 @@
+
+{
+      "terminal.integrated.env.osx": {
+            "PYTHONPATH": "${workspaceFolder}/"
+      },
+      "terminal.integrated.env.linux": {
+            "PYTHONPATH": "${workspaceFolder}/"
+      },
+      "python.testing.pytestArgs": [
+            "tests"
+      ],
+      "python.testing.unittestEnabled": false,
+      "python.testing.pytestEnabled": true,
+      "python.testing.cwd": "${workspaceFolder}/"
+}
diff --git a/requirements.txt b/requirements.txt
@@ -39,6 +39,7 @@ scipy==1.11.4
 shapely==2.0.2
 six==1.16.0
 tomli==2.0.1
-typing-extensions==4.9.0
+tqdm==4.66.1
+typing_extensions==4.9.0
 yarl==1.9.4
 zipp==3.16.2
diff --git a/spf/sdrpluto/old/01_phase_sync.py b/spf/sdrpluto/old/01_phase_sync.py
diff --git a/spf/sdrpluto/old/02_wifi_direction.py b/spf/sdrpluto/old/02_wifi_direction.py
diff --git a/spf/sdrpluto/phase_cal/01_phase_cal.py → ...ndalone_scripts/phase_cal/01_phase_cal.py b/spf/sdrpluto/phase_cal/01_phase_cal.py → ...ndalone_scripts/phase_cal/01_phase_cal.py
diff --git a/...drpluto/test_emitter_recv/03_only_emit.py → ...scripts/test_emitter_recv/03_only_emit.py b/...drpluto/test_emitter_recv/03_only_emit.py → ...scripts/test_emitter_recv/03_only_emit.py
diff --git a/...pluto/test_emitter_recv/04_plot_signal.py → ...ripts/test_emitter_recv/04_plot_signal.py b/...pluto/test_emitter_recv/04_plot_signal.py → ...ripts/test_emitter_recv/04_plot_signal.py
diff --git a/...to/test_single_phase/02_wifi_direction.py → ...ts/test_single_phase/02_wifi_direction.py b/...to/test_single_phase/02_wifi_direction.py → ...ts/test_single_phase/02_wifi_direction.py
diff --git a/spf/sdrpluto/test.py b/spf/sdrpluto/test.py
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
@@ -0,0 +1,56 @@
+import argparse
+import os
+import numpy as np
+import tempfile
+from spf.dataset.spf_generate import generate_session_and_dump, generate_session
+from compress_pickle import dump
+import pytest
+
+from joblib import Parallel, delayed
+
+class dotdict(dict):
+    """dot.notation access to dictionary attributes"""
+    __getattr__ = dict.get
+    __setattr__ = dict.__setitem__
+    __delattr__ = dict.__delitem__
+
+@pytest.fixture
+def default_args():
+    return dotdict({
+            'carrier_frequency':2.4e9,
+            'signal_frequency':100e3,
+            'sampling_frequency':10e6,
+            'array_type':"linear", # "circular"],
+            'elements':11,
+            'random_silence':False,
+            'detector_noise':1e-4,
+            'random_emitter_timing':False,
+            'sources':2,
+            'seed':0,
+            'beam_former_spacing':256 + 1,
+            'width':128,
+            'detector_trajectory':'bounce',
+            'detector_speed':10.0,
+            'source_speed':0.0,
+            'sigma':1.0,
+            'time_steps':100,
+            'time_interval':0.3,
+            'samples_per_snapshot':3,
+            'sessions':16,
+
+            'reference':False,
+            'cpus':8,
+            'live':False,
+            'profile':False,
+            'fixed_detector':None # 
+    })
+
+def test_data_generation(default_args):
+    with tempfile.TemporaryDirectory() as tmp:
+        args = default_args
+        args.output=tmp
+        print(args)
+
+        dump(args, "/".join([args.output, "args.pkl"]), compression="lzma")
+        result = [ generate_session_and_dump((args, session_idx))
+                for session_idx in range(args.sessions) ]