Changed project structure

airbrakes/airbrakes.py

@@ -2,11 +2,11 @@
 
 from typing import TYPE_CHECKING
 
-from airbrakes.imu.data_processor import IMUDataProcessor
-from airbrakes.imu.imu import IMU, IMUDataPacket
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket
-from airbrakes.logger import Logger
-from airbrakes.servo import Servo
+from airbrakes.data_handling.data_processor import IMUDataProcessor
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket
+from airbrakes.data_handling.logger import Logger
+from airbrakes.hardware.imu import IMU, IMUDataPacket
+from airbrakes.hardware.servo import Servo
 from airbrakes.state import StandByState, State
 
 if TYPE_CHECKING:

airbrakes/imu/data_processor.py → airbrakes/data_handling/data_processor.py

@@ -3,7 +3,7 @@
 import statistics as stats
 from collections.abc import Sequence
 
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket
 
 
 class IMUDataProcessor:

airbrakes/logger.py → airbrakes/data_handling/logger.py

@@ -5,8 +5,8 @@
 import multiprocessing
 from pathlib import Path
 
-from airbrakes.constants import CSV_HEADERS, STOP_SIGNAL
-from airbrakes.imu.imu_data_packet import IMUDataPacket
+from airbrakes.data_handling.imu_data_packet import IMUDataPacket
+from constants import CSV_HEADERS, STOP_SIGNAL
 
 
 class Logger:

airbrakes/imu/imu.py → airbrakes/hardware/imu.py

@@ -13,8 +13,8 @@
         stacklevel=2,
     )
 
-from airbrakes.constants import ESTIMATED_DESCRIPTOR_SET, MAX_QUEUE_SIZE, RAW_DESCRIPTOR_SET
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket, IMUDataPacket, RawDataPacket
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket, IMUDataPacket, RawDataPacket
+from constants import ESTIMATED_DESCRIPTOR_SET, MAX_QUEUE_SIZE, RAW_DESCRIPTOR_SET
 
 
 class IMU:

airbrakes/constants.py → constants.py

@@ -2,7 +2,7 @@
 
 from pathlib import Path
 
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket, IMUDataPacket, RawDataPacket
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket, IMUDataPacket, RawDataPacket
 
 # -------------------------------------------------------
 # Servo Configuration

main.py

@@ -2,10 +2,10 @@
 loop."""
 
 from airbrakes.airbrakes import AirbrakesContext
-from airbrakes.constants import FREQUENCY, LOGS_PATH, MAX_EXTENSION, MIN_EXTENSION, PORT, SERVO_PIN, UPSIDE_DOWN
-from airbrakes.imu.imu import IMU
-from airbrakes.logger import Logger
-from airbrakes.servo import Servo
+from airbrakes.data_handling.logger import Logger
+from airbrakes.hardware.imu import IMU
+from airbrakes.hardware.servo import Servo
+from constants import FREQUENCY, LOGS_PATH, MAX_EXTENSION, MIN_EXTENSION, PORT, SERVO_PIN, UPSIDE_DOWN
 
 
 def main():

pyproject.toml

@@ -6,6 +6,7 @@ version = "0.1.0"
 dependencies = [
     "gpiozero",
     "pigpio",  # Run sudo pigpiod before running the program
+    "pytest",
     # Installation instructions for the following dependencies can be found in the README:
     # "mscl" https://github.com/LORD-MicroStrain/MSCL/blob/master/BuildScripts/buildReadme_Linux.md
 ]

scripts/run_imu.py

@@ -4,13 +4,11 @@
 For the pi, you will have to use python3
 """
 
-from airbrakes.constants import FREQUENCY, PORT, UPSIDE_DOWN
-from airbrakes.imu.imu import IMU
-from airbrakes.logger import Logger
+from constants import FREQUENCY, PORT, UPSIDE_DOWN
+from airbrakes.hardware.imu import IMU
+from airbrakes.data_handling.logger import Logger
 from pathlib import Path
 
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket, RawDataPacket
-
 # import matplotlib.pyplot as plt
 # import matplotlib.animation as animation
 # from collections import deque

scripts/run_logger.py

@@ -3,9 +3,9 @@
 import time
 from collections import deque
 
-from airbrakes.constants import LOGS_PATH
-from airbrakes.imu.imu_data_packet import RawDataPacket
-from airbrakes.logger import Logger
+from constants import LOGS_PATH
+from airbrakes.data_handling.imu_data_packet import RawDataPacket
+from airbrakes.data_handling.logger import Logger
 
 
 def main():

scripts/run_servo.py

@@ -4,8 +4,8 @@
 For the pi, you will have to use python3
 """
 
-from airbrakes.constants import MAX_EXTENSION, MIN_EXTENSION, SERVO_PIN
-from airbrakes.servo import Servo
+from constants import MAX_EXTENSION, MIN_EXTENSION, SERVO_PIN
+from airbrakes.hardware.servo import Servo
 
 servo = Servo(SERVO_PIN, MIN_EXTENSION, MAX_EXTENSION)
 

tests/conftest.py

@@ -5,10 +5,10 @@
 import pytest
 from gpiozero.pins.mock import MockFactory, MockPWMPin
 
-from airbrakes.constants import FREQUENCY, MAX_EXTENSION, MIN_EXTENSION, PORT, SERVO_PIN, UPSIDE_DOWN
-from airbrakes.imu.imu import IMU
-from airbrakes.logger import Logger
-from airbrakes.servo import Servo
+from airbrakes.data_handling.logger import Logger
+from airbrakes.hardware.imu import IMU
+from airbrakes.hardware.servo import Servo
+from constants import FREQUENCY, MAX_EXTENSION, MIN_EXTENSION, PORT, SERVO_PIN, UPSIDE_DOWN
 
 LOG_PATH = Path("tests/logs")
 

tests/test_data_processor.py

@@ -3,8 +3,8 @@
 
 import pytest
 
-from airbrakes.imu.data_processor import IMUDataProcessor
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket
+from airbrakes.data_handling.data_processor import IMUDataProcessor
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket
 
 
 def simulate_altitude_sine_wave(n_points=1000, frequency=0.01, amplitude=100, noise_level=3, base_altitude=20):

tests/test_imu.py

@@ -3,9 +3,9 @@
 import time
 from collections import deque
 
-from airbrakes.constants import FREQUENCY, PORT, UPSIDE_DOWN
-from airbrakes.imu.imu import IMU
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket, IMUDataPacket, RawDataPacket
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket, IMUDataPacket, RawDataPacket
+from airbrakes.hardware.imu import IMU
+from constants import FREQUENCY, PORT, UPSIDE_DOWN
 
 RAW_DATA_PACKET_SAMPLING_RATE = 1 / 1000  # 1kHz
 EST_DATA_PACKET_SAMPLING_RATE = 1 / 500  # 500Hz

tests/test_imu_data_packet.py

@@ -1,6 +1,6 @@
 import pytest
 
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket, RawDataPacket
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket, RawDataPacket
 
 
 class TestEstimatedDataPacket:

tests/test_logger.py

@@ -5,9 +5,9 @@
 
 import pytest
 
-from airbrakes.constants import CSV_HEADERS
-from airbrakes.imu.imu_data_packet import EstimatedDataPacket, RawDataPacket
-from airbrakes.logger import Logger
+from airbrakes.data_handling.imu_data_packet import EstimatedDataPacket, RawDataPacket
+from airbrakes.data_handling.logger import Logger
+from constants import CSV_HEADERS
 from tests.conftest import LOG_PATH