1
0
mirror of https://github.com/undera/pylgbst.git synced 2020-11-18 19:37:26 -08:00

Working on tilt sensor

This commit is contained in:
Andrey Pohilko 2017-09-14 10:08:42 +03:00
parent 3e1c5c4892
commit 8c1e8f61c5
3 changed files with 22 additions and 13 deletions

View File

@ -29,6 +29,7 @@ for device in hub.devices:
- Make it 2/3 compatible
- Add travis unit tests and coverage
- Give nice documentation examples, don't forget to mention logging
- make angled motors to be synchronous by default
## Links

View File

@ -10,10 +10,15 @@ class Peripheral(object):
"""
def __init__(self, parent, port):
"""
:type parent: pylgbst.MoveHub
:type port: int
"""
super(Peripheral, self).__init__()
self.parent = parent
self.port = port
self.working = False
self._subscribers = []
def __repr__(self):
return "%s on port %s" % (self.__class__.__name__, PORTS[self.port] if self.port in PORTS else 'N/A')
@ -117,8 +122,10 @@ class ColorDistanceSensor(Peripheral):
class TiltSensor(Peripheral):
def subscribe(self, callback):
params = b'\x00\x01\x00\x00\x00\x01' # full
# params = b'\x02\x01\x00\x00\x00\x01' # basic
self._subscribe_on_port(params)
self._subscribers.append(callback)
self._subscribers.append(callback) # TODO: maybe join it into `_subscribe_on_port`
class Button(Peripheral):

25
test.py
View File

@ -3,9 +3,10 @@ import time
import unittest
from threading import Thread
from pylgbst import MoveHub, COLOR_RED, LED, EncodedMotor, PORT_AB, Peripheral
from pylgbst import MoveHub, COLOR_RED, LED, EncodedMotor, PORT_AB
from pylgbst.comms import Connection, str2hex, hex2str
from pylgbst.constants import PORT_LED
from pylgbst.constants import PORT_LED, PORT_TILT_SENSOR
from pylgbst.peripherals import TiltSensor
logging.basicConfig(level=logging.DEBUG)
@ -51,29 +52,29 @@ class ConnectionMock(Connection):
class HubMock(MoveHub):
def __init__(self, connection=None):
super(HubMock, self).__init__(connection if connection else ConnectionMock())
def _wait_for_devices(self):
pass
class GeneralTest(unittest.TestCase):
def test_led(self):
conn = ConnectionMock()
conn.notifications.append((14, '1b0e00 0900 04 39 0227003738'))
hub = HubMock(conn)
hub = HubMock()
led = LED(hub, PORT_LED)
led.set_color(COLOR_RED)
self.assertEquals("0701813211510009", conn.writes[0][1])
self.assertEquals("0801813211510009", hub.connection.writes[0][1])
def test_tilt_sensor(self):
conn = ConnectionMock()
conn.notifications.append((14, '1b0e000f00043a0128000000000100000001'))
hub = HubMock(conn)
hub = HubMock()
hub.tilt_sensor = TiltSensor(hub, PORT_TILT_SENSOR)
def callback():
pass
hub.tilt_sensor.subscribe(callback)
self.assertEquals("0701813211510009", conn.writes[1][1])
self.assertEquals("0a01413a000100000001", hub.connection.writes[0][1])
def test_motor(self):
conn = ConnectionMock()
@ -81,9 +82,9 @@ class GeneralTest(unittest.TestCase):
hub = HubMock(conn)
motor = EncodedMotor(hub, PORT_AB)
motor.timed(1.5)
self.assertEquals("0c018139110adc056464647f03", conn.writes[0][1])
self.assertEquals("0d018139110adc056464647f03", conn.writes[0][1])
motor.angled(90)
self.assertEquals("0e018139110c5a0000006464647f03", conn.writes[1][1])
self.assertEquals("0f018139110c5a0000006464647f03", conn.writes[1][1])
def test_capabilities(self):
conn = ConnectionMock()