wrdk/testudo/tests/test_SAMO_01.py

# test main board
# -*- coding: utf-8 -*-
# COPYRIGHT: Openmoko Inc. 2009
# LICENSE: GPL Version 2 or later
# DESCRIPTION: Main board test
# AUTHOR: Christopher Hall <hsw@openmoko.com>

import Keithley
import Agilent
import RelayBoard
import communication
import process

import sys
import time


if 'serial_number' in global_args:
    serial_number = global_args['serial_number']
else:
    serial_number = 'No Serial Number'


psu = None
dvm = None
relay = None

RELAY_SERIAL = '/dev/USBrelay'
CPU_SERIAL = '/dev/USBconsole'

RELAY_VBATT = 1
RELAY_RESET = 2
RELAY_POWER_SWITCH = 3
RELAY_RANDOM_KEY = 4
RELAY_HISTORY_KEY = 5
RELAY_SEARCH_KEY = 6
RELAY_RXD  = 7
RELAY_TXD = 8
RELAY_PROGRAM_FLASH = 9
RELAY_1V8 = 10
RELAY_3V = 11
RELAY_LCD_V0 = 12
RELAY_LCD_V1 = 13
RELAY_LCD_V2 = 14
RELAY_LCD_V3 = 15
RELAY_LCD_V4 = 16

# volts
LCD_V0 = 21.0
LCD_V0_DELTA = 0.1

# power supply (volts, amps)
SUPPLY_STANDARD_VOLTAGE = 3.0
SUPPLY_CURRENT_LIMIT = 0.35

# specify the voltages actual and +/- percentages
# or minimum/maximum
VOLTAGE_LIST = (
#   ("text", relay_name,   required_value,       percent_low, percent_high)
#   ("text", relay_name,   None,                 minimum, maximum)
    ("1V8",  RELAY_1V8,    1.8,                   -5.0,  5.0),
    ("3V ",  RELAY_3V,     None,                   3.2,  3.4),
    ("V0 ",  RELAY_LCD_V0, LCD_V0,               -10.0, 10.0),
    ("V1 ",  RELAY_LCD_V1, LCD_V0 * 14.0 / 15.0, -10.0, 10.0),
    ("V2 ",  RELAY_LCD_V2, LCD_V0 * 13.0 / 15.0, -10.0, 10.0),
    ("V3 ",  RELAY_LCD_V3, LCD_V0 *  2.0 / 15.0, -10.0, 10.0),
    ("V4 ",  RELAY_LCD_V4, LCD_V0 *  1.0 / 15.0, -10.0, 10.0),
)

# amps
MAXIMUM_LEAKAGE_CURRENT = 0.002
MINIMUM_ON_CURRENT = 0.012
MAXIMUM_ON_CURRENT = 0.150

# seconds
MINIMUM_ON_TIME = 0.01
MAXIMUM_ON_TIME = 1.2
MINIMUM_OFF_TIME = 1.7
MAXIMUM_OFF_TIME = 4.0
ON_OFF_DELTA = 0.01
AVERAGING_DELAY = 0.02
SETTLING_TIME = 0.25
VOLTAGE_SAMPLE_TIME = 0.02
VOLTAGE_SAMPLE_OFF = 0.01
RESET_TIME = 0.5

# 1/10 seconds
ON_OFF_SCAN = int(5 / ON_OFF_DELTA)

# function keys
KEY_LIST = [
    ('Random',  RELAY_RANDOM_KEY,  '0x02'),
    ('History', RELAY_HISTORY_KEY, '0x04'),
    ('Search',  RELAY_SEARCH_KEY,  '0x01')
]


def setUp():
    """Set up power supply and turn on

       Also put a message on the PSU LCD to warn operator"""
    global debug, psu, dvm, relay
    info('setUp: **initialising**')

    relay = RelayBoard.PIC16F873A(port = RELAY_SERIAL)

    dvm = Agilent.DMM34401A()
    dvm.setVoltageDC()

    psu = Keithley.PSU2303()
    psu.setCurrent(SUPPLY_CURRENT_LIMIT)
    psu.setVoltage(SUPPLY_STANDARD_VOLTAGE)
    psu.powerOff()
    if debug:
        psu.settings()
        psu.measure()
    psu.message('Test in progress  Do NOT Touch  ')
    relay.on(RELAY_VBATT)


def tearDown():
    """Shutdown the power supply"""
    global debug, psu, dvm, relay
    relay.off(RELAY_VBATT)
    psu.powerOff()
    psu.setCurrent(0)
    psu.setVoltage(0)
    psu.messageOff()
    info('tearDown: **cleanup**')
    del psu
    psu = None
    del dvm
    dvm = None
    del relay
    relay = None


def test001_leakage():
    """Make sure power is off and no leakage"""
    global debug, psu, dvm, relay
    relay.off(RELAY_POWER_SWITCH)
    time.sleep(RESET_TIME)
    psu.powerOn()
    time.sleep(SETTLING_TIME)
    if debug:
        psu.settings()
        psu.measure()
    i = psu.current
    info('Leakage current = %7.3f mA @ %5.1f V' % (1000 * i, psu.voltage))
    fail_if(abs(i) > MAXIMUM_LEAKAGE_CURRENT, "Leakage current %7.3f mA is too high" % (i * 1000))


def test002_on():
    """Turn on power and wait for current to rise"""
    global debug, psu, dvm, relay
    relay.set(RELAY_PROGRAM_FLASH)
    relay.set(RELAY_POWER_SWITCH)
    relay.update()
    t = time.time()
    for i in range(ON_OFF_SCAN):
        if psu.current >= MINIMUM_ON_CURRENT:
            break
        time.sleep(ON_OFF_DELTA)
    t = time.time() - t
    relay.off(RELAY_POWER_SWITCH)
    time.sleep(SETTLING_TIME)
    info('On current = %7.3f mA @ %5.1f V' % (1000 * psu.current, psu.voltage))
    fail_unless(psu.current >= MINIMUM_ON_CURRENT, "Failed to Power On")
    fail_if(t < MINIMUM_ON_TIME, "On too short, %5.1f s < %5.1f" % (t, MINIMUM_ON_TIME))
    fail_if(t > MAXIMUM_ON_TIME, "On too long, %5.1f s > %5.1f" % (t, MAXIMUM_ON_TIME))


def test003_check_current():
    """Monitor current to check if correct"""
    global debug, psu, dvm, relay
    averageCurrent = 0.0
    samples = 20
    for i in range(samples):
        if debug:
            psu.measure()
        time.sleep(AVERAGING_DELAY)
        i = psu.current
        averageCurrent = averageCurrent + i
        info('Supply current = %7.3f mA @ %5.1f V' % (1000 * i, psu.voltage))
        fail_unless(abs(i) > MINIMUM_ON_CURRENT, "Device failed to power up")
        fail_if(abs(i) > MAXIMUM_ON_CURRENT, "Device current too high")
    info('Average supply current = %7.3f mA @ %5.1f V' % (1000 * averageCurrent / samples, psu.voltage))


def test004_measure_voltages():
    """Measure voltages"""
    global debug, psu, dvm, relay
    for item in VOLTAGE_LIST:
        v = item[0]
        r = item[1]
        if None == item[2]:
            min = item[3]
            max = item[4]
        else:
            min = item[2] * (100 + item[3]) / 100
            max = item[2] * (100 + item[4]) / 100
        relay.on(r)
        time.sleep(VOLTAGE_SAMPLE_TIME)
        actual = dvm.voltage
        info('%s = %7.3f V' % (v, actual))
        fail_if(actual < min, "Low Voltage %s = %7.3f < %7.3f" % (v, actual, min))
        fail_if(actual > max, "High Voltage %s = %7.3f > %7.3f" % (v, actual, max))
        relay.off(r)
        time.sleep(VOLTAGE_SAMPLE_OFF)


def test005_power_off():
    """Check power off function"""
    global debug, psu, dvm, relay
    relay.clear(RELAY_PROGRAM_FLASH)
    relay.set(RELAY_POWER_SWITCH)
    relay.update()
    t = time.time()
    for i in range(ON_OFF_SCAN):
        if psu.current < MINIMUM_ON_CURRENT:
            break
        time.sleep(ON_OFF_DELTA)
    t = time.time() - t
    relay.off(RELAY_POWER_SWITCH)
    time.sleep(2)
    i = psu.current
    fail_unless(abs(i) < MAXIMUM_LEAKAGE_CURRENT, "Failed to power off , %7.3f mA" % (i * 1000))
    fail_if(t < MINIMUM_OFF_TIME, "Off too short, %5.1f s < %5.1f" % (t, MINIMUM_OFF_TIME))
    fail_if(t > MAXIMUM_OFF_TIME, "Off too long, %5.1f s > %5.1f" % (t, MAXIMUM_OFF_TIME))


def test006_on():
    """Turn on power and wait for current to rise"""
    global debug, psu, dvm, relay
    test002_on()


def test007_program_flash():
    """Program the boot loader into FLASH memory"""
    global debug, psu, dvm, relay, flash_tries
    relay.set(RELAY_RESET)
    relay.set(RELAY_PROGRAM_FLASH)
    relay.set(RELAY_RXD)
    relay.set(RELAY_TXD)
    relay.update()

    def callback(s):
        global debug, psu, dvm, relay, flash_tries
        i = psu.current
        info('Supply current = %7.3f mA' % (1000 * i))
        fail_if(0 == flash_tries, 'Too many reset attempts (No CPU response)')
        info(s.replace('\10', ''))  # remove backspaces
        if 'Press Reset' == s.strip():
            info('Resets remaining: %d' % flash_tries)
            relay.on(RELAY_RESET)
            time.sleep(RESET_TIME)
            relay.off(RELAY_RESET)
            flash_tries -= 1

    flash_tries = 10
    p = process.Process(['make',
                         'flash-mbr',
                         'FLASH_UPDATE=YES',
                         'BOOTLOADER_TTY=' + CPU_SERIAL,
                         "SERIAL_NUMBER='" + serial_number + "'"], callback)

    rc = p.run()
    fail_unless(rc, 'Flashing failed')
    relay.off(RELAY_PROGRAM_FLASH)


def test008_internal():
    """Run internal test program

    Memory Check
    Test three function keys
    Calibrate LCD contrast voltage
    Check software power off"""

    global debug, psu, dvm, relay
    p = communication.SerialPort(port = CPU_SERIAL)

    relay.on(RELAY_RESET)
    relay.clear(RELAY_PROGRAM_FLASH)
    relay.clear(RELAY_RANDOM_KEY)
    relay.clear(RELAY_HISTORY_KEY)
    relay.clear(RELAY_SEARCH_KEY)
    relay.set(RELAY_RXD)
    relay.set(RELAY_TXD)
    relay.update()
    time.sleep(RESET_TIME)
    relay.off(RELAY_RESET)

    m_rev = p.waitFor(r'[rR][eE][vV]:\s*(\S+)\s')
    if m_rev:
        info('Hardware ' + m_rev.group(0))

    fail_unless(p.waitFor(r'menu\?'), 'boot loader failed to start')
    p.send(' ')

    m_mem = p.waitFor(r'(.)\.\s+[mM]emory\s+[cC]heck')
    fail_unless(m_mem, 'Boot Loader missing Memory Check option')

    m_key = p.waitFor(r'(.)\.\s+[kK]ey\s+[tT]est')
    fail_unless(m_key, 'Boot Loader missing Key Test option')

    fail_unless(p.waitFor(r'[sS]election:'), 'Boot Loader menu prompt failed')
    p.send(m_mem.group(1))

    m_mem = p.waitFor(r'[mM]emory:[^\]]+\]')
    fail_unless(m_mem, 'Memory Check did not respond')
    info(m_mem.group(0))

    m_mem = p.waitFor(r'(PASS|FAIL):\s+(.*)\n')
    fail_unless(m_mem, 'Memory Check did not respond')
    fail_unless('PASS' == m_mem.group(1), m_mem.group(2))
    info(m_mem.group(1) + ': ' + m_mem.group(2))

    fail_unless(p.waitFor(r'[sS]election:'), 'Boot Loader menu prompt failed')
    p.send(m_key.group(1))

    for desc, r, k in KEY_LIST:
        relay.off(r)
        fail_unless(p.waitFor('keys = '), 'Key Test did not respond')
        key = p.read(4)
        info('key (none) = %s' % key)
        fail_unless('0x00' == key, 'Invalid keys: wanted %s, got %s' % ('0x00', key))
        relay.on(r)
        fail_unless(p.waitFor('keys = '), 'Key Test did not respond')
        key = p.read(4)
        i = psu.current
        info('Supply current = %7.3f mA' % (1000 * i))
        info('key (%s)[%s] = %s' % (desc, k, key))
        fail_unless(k == key, 'Invalid keys: wanted %s, got %s' % (k, key))
        relay.off(r)

    fail_unless(p.waitFor('keys = '), 'Key Test did not respond')
    key = p.read(4)
    info('key (none) = %s' % key)
    fail_unless('0x00' == key, 'Invalid keys: wanted %s, got %s' % ('0x00', key))

    # exit key test and wait for prompt
    # contrast control should then be active
    p.send('\n')
    fail_unless(p.waitFor(r'[sS]election:'), 'Boot Loader menu prompt failed')

    info('Calibrate LCD Voltages to %7.3f V +- %7.3f V' % (LCD_V0, LCD_V0_DELTA))
    relay.on(RELAY_LCD_V0)
    relay_decrease = RELAY_RANDOM_KEY
    relay_increase = RELAY_SEARCH_KEY
    relay_set = RELAY_HISTORY_KEY

    v0_max = LCD_V0 + LCD_V0_DELTA
    v0_min = LCD_V0 - LCD_V0_DELTA
    actual = 0
    for i in range(20):
        time.sleep(VOLTAGE_SAMPLE_TIME)
        actual = dvm.voltage
        #info('Contrast voltage = %7.3f V' % actual)
        if actual > v0_max:
            relay.set(relay_decrease)
            relay.off(relay_increase)
        elif actual < v0_min:
            relay.set(relay_increase)
            relay.off(relay_decrease)
        else:
            relay.set(relay_set)
            relay.clear(relay_increase)
            relay.off(relay_decrease)
            break

    relay.clear(relay_increase)
    relay.clear(relay_decrease)
    relay.off(RELAY_LCD_V0)
    time.sleep(VOLTAGE_SAMPLE_TIME)
    relay.off(relay_set)

    fail_if(actual > v0_max or actual < v0_min,
            'LCD contrast voltage: %f7.3 out of range: %7.3f .. %7.3f' % (actual, v0_min, v0_max))

    info('Calibrate LCD Voltages Completed; New values are:')
    test004_measure_voltages()

    info('sending auto power off sequence')
    p.send('\n0\n')
    del p
    p = None

    for n in range(5):
        time.sleep(SETTLING_TIME)
        i = psu.current
        info('Supply current = %7.3f mA' % (1000 * i))
        if abs(i) < MAXIMUM_LEAKAGE_CURRENT:
            break
    fail_if(abs(i) > MAXIMUM_LEAKAGE_CURRENT, "Failed auto power off, current %7.3f mA is too high" % (i * 1000))


#def test009_power_off():
#    """Check power off function"""
#    global debug, psu, dvm, relay
#    test005_power_off()