test.py

import time
import urllib

import numpy as np

from stbt import press, wait_until, match, Region


black_rect = np.empty((8, 8, 3), dtype=np.uint8)
black_rect.fill(16)
region = Region(0, 0, width=8, height=8)

def test_motorola():
    while True:
        start = time.time()
        press('KEY_LEFT')
        try:
            assert wait_until(lambda: match(black_rect, region=region))
        except AssertionError:
            send_result(0, 0)
        else:
            end = time.time()
            send_result(start, end)
        finally:
            teardown()


def send_result(start_time, end_time):
    urllib.urlopen('http://172.17.80.225:5000?time={0}'.format(end_time - start_time))


def teardown():
    press('KEY_RIGHT')