fix: basic_test

Author: zlj-zz

tests/basic_test.py

@@ -1,5 +1,4 @@
 import time
-import random
 import os
 from pymycobot.mycobot import MyCobot
 from pymycobot.genre import Angle, Coord
@@ -10,11 +9,8 @@
 def test(mycobot):
     print("\nStart check basic options\n")
 
-    color_name = ["red", "green", "blue"]
-    color_code = ["ff0000", "00ff00", "0000ff"]
-    i = random.randint(0, len(color_code) - 1)
-    mycobot.set_led_color(color_code[i])
-    print("::set_led_color() ==> color {}\n".format(color_name[i]))
+    mycobot.set_color(255, 255, 0)
+    print("::set_color() ==> color {}\n".format("255 255 0"))
     time.sleep(3)
 
     angles = [0, 0, 0, 0, 0, 0]
@@ -39,7 +35,8 @@ def test(mycobot):
 
     coords = [160, 160, 160, 0, 0, 0]
     mycobot.send_coords(coords, 70, 0)
-    print("::send_coords() ==> send coords {}, speed 70, mode 0\n".format(coords))
+    print("::send_coords() ==> send coords {}, speed 70, mode 0\n".format(
+        coords))
     time.sleep(3)
 
     print("::get_coords() ==> coords {}\n".format(mycobot.get_coords()))
@@ -58,8 +55,7 @@ def test(mycobot):
 
 
 if __name__ == "__main__":
-    print(
-        """
+    print("""
 --------------------------------------------
 | This file will test basic option method: |
 |     set_led_color()                      |
@@ -73,8 +69,7 @@ def test(mycobot):
 |     send_coord()                         |
 |     set_free_mode()                      |
 --------------------------------------------
-          """
-    )
+          """)
     time.sleep(3)
     # port = subprocess.check_output(['echo -n /dev/ttyUSB*'],
     # shell=True).decode()

tests/gripper_test.py

@@ -1,4 +1,5 @@
-import time, subprocess
+import os
+import time
 from pymycobot.mycobot import MyCobot
 
 
@@ -10,9 +11,11 @@ def gripper_test(mc):
     print("Is gripper moving: {}".format(flag))
     time.sleep(1)
 
-    print("Set the current position to zero")
-    mc.set_gripper_ini()
-    time.sleep(2)
+    # Set the current position to (2048).
+    # Use it when you are sure you need it.
+    # Gripper has been initialized for a long time. Generally, there
+    # is no need to change the method.
+    # mc.set_gripper_ini()
 
     print("")
     mc.set_gripper_value(2048, 50)
@@ -35,6 +38,8 @@ def gripper_test(mc):
 
 
 if __name__ == "__main__":
-    port = subprocess.check_output(["echo -n /dev/ttyUSB*"], shell=True).decode()
+    with open(os.path.dirname(__file__) + "/port.txt") as f:
+        port = f.read().strip().replace("\n", "")
+        print(port)
     mycobot = MyCobot(port)
     gripper_test(mycobot)

tests/io_test.py

@@ -1,4 +1,5 @@
-import time, subprocess
+import os
+import time
 from pymycobot.mycobot import MyCobot
 
 
@@ -14,6 +15,8 @@ def io_test(mc):
 
 
 if __name__ == "__main__":
-    port = subprocess.check_output(["echo -n /dev/ttyUSB*"], shell=True).decode()
+    with open(os.path.dirname(__file__) + "/port.txt") as f:
+        port = f.read().strip().replace("\n", "")
+        print(port)
     mycobot = MyCobot(port)
     io_test(mycobot)

tests/port.txt

@@ -1 +1 @@
-/dev/cu.usbserial-7152ADB1F6
+/dev/cu.usbserial-0213245D