nilmdb/tests/test_helpers.py

# Just some helpers for test functions

import shutil, os

def eq_(a, b):
    if not a == b:
        raise AssertionError("%r != %r" % (a, b))

def in_(a, b):
    if a not in b:
        raise AssertionError("%r not in %r" % (a, b))

def ne_(a, b):
    if not a != b:
        raise AssertionError("unexpected %r == %r" % (a, b))

def recursive_unlink(path):
    try:
        shutil.rmtree(path)
    except OSError:
        pass
    try:
        os.unlink(path)
    except OSError:
        pass