test_datadiff_tools.py

import sys
from textwrap import dedent

from datadiff import tools

from test_datadiff import assert_equal


def test_assert_equal_true():
    # nothing raised
    assert_equal(None, tools.assert_equals(7, 7))
    
def test_assert_equal_false():
    try:
        tools.assert_equals([3,4], [5,6])
    except:
        e = sys.exc_info()[1]
        assert_equal(type(e), AssertionError)
        assert_equal(str(e), dedent('''\
            
            --- a
            +++ b
            [
            @@ -0,1 +0,1 @@
            -3,
            -4,
            +5,
            +6,
            ]'''))
    else:
        raise AssertionError("Should've raised an AssertionError")
    
def test_assert_equal_msg():
    try:
        tools.assert_equals(3, 4, "whoops")
    except:
        e = sys.exc_info()[1]
        assert_equal(type(e), AssertionError,
                     "Raised exception should be AssertionError")
        assert_equal(str(e), "whoops")
    else:
        raise AssertionError("Should've raised an AssertionError")
    
def test_assert_equals():
    assert_equal(tools.assert_equal, tools.assert_equals)

def test_assert_equal_simple():
    try:
        tools.assert_equals(True, False)
    except:
        e = sys.exc_info()[1]
        assert_equal(type(e), AssertionError)
        assert_equal(str(e), dedent('''\
            True != False'''))
    else:
        raise AssertionError("Should've raised an AssertionError")

def test_assert_equal_simple_types():
    try:
        tools.assert_equals('a', 7)
    except:
        e = sys.exc_info()[1]
        assert_equal(type(e), AssertionError)
        assert_equal(str(e), dedent('''\
            'a' != 7'''))
    else:
        raise AssertionError("Should've raised an AssertionError")


if __name__ == '__main__':
    try:
        import nose
    except (ImportError, SyntaxError, NameError):
        pass
    else:
        nose.main()
        sys.exit(0)
    
    import types
    for fn_name, fn in sorted(locals().items()):
        if fn_name.startswith('test_') and type(fn) == types.FunctionType:
            sys.stderr.write(fn_name + '...\n')
            fn()