update test file

Author: zlj-zz

tests/basic_test.py

@@ -0,0 +1,80 @@
+import time, random, subprocess
+from pymycobot.mycobot import MyCobot
+from pymycobot.genre import Angle, Coord
+
+reset = [153.19, 137.81, -153.54, 156.79, 87.27, 13.62]
+
+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]))
+    time.sleep(3)
+
+    angles = [0,0,0,0,0,0]
+    mycobot.send_angles(angles, 100)
+    print('::send_angles() ==> angles {}, speed 100\n'.format(angles))
+    time.sleep(3)
+
+    print('::get_angles() ==> degrees: {}\n'.format(mycobot.get_angles()))
+    time.sleep(1)
+
+    mycobot.send_angle(Angle.J1.value, 90, 50)
+    print('::send_angle() ==> angle: joint1, degree: 90, speed: 50\n')
+    time.sleep(4)
+
+    radians = [1,1,1,1,1,1]
+    mycobot.send_radians(radians, 100)
+    print('::send_radians() ==> set radians {}, speed 100\n'.format(radians))
+    time.sleep(3)
+
+    print('::get_radians() ==> radians: {}\n'.format(mycobot.get_radians()))
+    time.sleep(1)
+
+    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))
+    time.sleep(3)
+
+    print('::get_coords() ==> coords {}\n'.format(mycobot.get_coords()))
+    time.sleep(0.5)
+
+    mycobot.send_coord(Coord.X.value, -40, 70)
+    print('::send_coord() ==> send coord id: X, coord value: -40, speed: 70\n')
+    time.sleep(2)
+
+    print('::set_free_mode()\n')
+    mycobot.send_angles(reset, 100)
+    time.sleep(5)
+    mycobot.set_free_mode()
+
+    print('=== check end ===\n')
+
+
+if __name__ == '__main__':
+    print('''
+————————————————————————————————————————————
+| This file will test basic option method: |
+|     set_led_color()                      |
+|     send_angles()                        |
+|     get_angles()                         |
+|     send_angle()                         |
+|     send_radians()                       |
+|     get_radians()                        |
+|     send_coords()                        |
+|     get_coords()                         |
+|     send_coord()                         |
+|     set_free_mode()                      |
+————————————————————————————————————————————
+          ''')
+    time.sleep(3)
+    # port = subprocess.check_output(['echo -n /dev/ttyUSB*'], 
+                                    # shell=True).decode()
+    port = "/dev/cu.usbserial-0203B030"
+    # mycobot = MyCobot(port, debug=True)
+    mycobot = MyCobot(port)
+    test(mycobot)
+

tests/loop_test.py

@@ -8,10 +8,9 @@
     # port = "/dev/cu.usbserial-0203B030"
     port = "/dev/cu.usbserial-0213245D"
     cobot = MyCobot(port)
-    # test(cobot)
     cobot.send_angles([0,0,0,0,0,0],100)
     time.sleep(10)
-    print('over')
+    print('start')
     for count in range(50):
         time.sleep(0.05)
         cobot.send_angle(1, (-30), 100)

tests/power_state_test.py

@@ -0,0 +1,62 @@
+import time, random, subprocess
+from pymycobot.mycobot import MyCobot
+from pymycobot.genre import Angle, Coord
+
+
+reset = [153.19, 137.81, -153.54, 156.79, 87.27, 13.62]
+zero = [0,0,0,0,0,0]
+sp = 100
+
+def test(mc):
+    print('Start check power state api\n')
+
+    mc.power_on()
+    time.sleep(.5)
+    f = mc.is_power_on()
+    print('is power on: {}'.format(f))
+    assert  f == 1, 'should be 1'
+    mc.send_angles(zero, sp)
+    time.sleep(3)
+
+    for _ in range(3):
+        mc.power_off()
+        time.sleep(.5)
+        f = mc.is_power_on()
+        print('is power on: {}'.format(f))
+        assert  f == 0, 'should be 0'
+
+        mc.send_angles(reset, sp)
+        time.sleep(1)
+
+        mc.power_on()
+        time.sleep(.5)
+        f = mc.is_power_on()
+        print('is power on: {}'.format(f))
+        assert  f == 1, 'should be 1'
+
+    mc.send_angles(reset, 100)
+    time.sleep(5)
+    mc.set_free_mode()
+
+    print('=== check end ===\n')
+
+
+
+
+
+if __name__ == '__main__':
+    print('''
+————————————————————————————————————————————
+| This file will test power state method:  |
+|     power_off()                          |
+|     power_on()                           |
+|     is_power_on()                        |
+————————————————————————————————————————————
+          ''')
+    time.sleep(3)
+    port = "/dev/cu.usbserial-0203B030"
+    mc = MyCobot(port)
+    test(mc)
+
+
+