Refactor propeller test to work when run directly.

chrisl8 · Oct 13, 2024 · 94b76c9 · 94b76c9
1 parent 2495d27
commit 94b76c9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/PropellerSerialTest/PropellerSerialTest.py → ...ot_ros/arlobot_ros/PropellerSerialTest.py b/PropellerSerialTest/PropellerSerialTest.py → ...ot_ros/arlobot_ros/PropellerSerialTest.py
@@ -21,10 +21,10 @@
 import curses
 import threading
 from math import sqrt, pow, radians, pi
-from PropellerSerialInterface import PropellerSerialInterface
-from PropellerSerialDataPacketTypes import PropellerSerialDataPacketTypes
-from PropellerSerialTestCursesInterface import Screen
-from checkPropellerCodeVersionNumber import checkPropellerCodeVersionNumber
+from . PropellerSerialInterface import PropellerSerialInterface
+from . PropellerSerialDataPacketTypes import PropellerSerialDataPacketTypes
+from . PropellerSerialTestCursesInterface import Screen
+from . checkPropellerCodeVersionNumber import checkPropellerCodeVersionNumber
 
 
 class Linear(object):

diff --git a/...est/PropellerSerialTestCursesInterface.py → ...ros/PropellerSerialTestCursesInterface.py b/...est/PropellerSerialTestCursesInterface.py → ...ros/PropellerSerialTestCursesInterface.py
diff --git a/scripts/PropellerSerialTest.sh b/scripts/PropellerSerialTest.sh
@@ -45,4 +45,5 @@ source "${HOME}/ros2_ws/install/setup.bash"
 
 export PYTHONPATH=${HOME}/ArloBot/arlobot_ros/arlobot_ros
 
-python "${SCRIPTDIR}/../PropellerSerialTest/PropellerSerialTest.py" "${USB_PORT}"
+cd "${SCRIPTDIR}/../arlobot_ros" || exit
+python -m arlobot_ros.PropellerSerialTest "${USB_PORT}"