# test_embedding.py
# Lillian Lee (LJL2@cornell.edu)
# Feb 26, 2013

import cunittest2
import embedding

def test_embed():
    """test embedding.embed(), on atoms and lists"""
    cunittest2.assert_equals(0, embedding.embed('a'))

    cunittest2.assert_equals(1, embedding.embed(['oh', 'hi']))
    cunittest2.assert_equals(1, embedding.embed(['oh', 'hi', 'there']))

    cunittest2.assert_equals(2, embedding.embed(['oh', ['hi'], 'there']))

    s1 = ['hit', ['a', 'guy'], ['with', 'glasses']]
    s2 = ['hit', ['a', 'guy',  ['with', 'glasses']]]

    cunittest2.assert_equals(2, embedding.embed(s1))
    cunittest2.assert_equals(3, embedding.embed(s2))
    s3 = ['the', [['red', 'house'], 'and', 'barn', ['that', 'jack', 'built']], 'was', 'razed']
    cunittest2.assert_equals(3, embedding.embed(s3))

    d1 = ['i', ['saw', ['her', 'duck'], ['with', 'a', 'telescope']]]
    d2 = ['i', ['saw', ['her', ['duck', ['with', 'a', 'telescope']]]]]
    cunittest2.assert_equals(3, embedding.embed(d1))
    cunittest2.assert_equals(5, embedding.embed(d2))


if __name__ == '__main__':
    test_embed()
    print "All embedding test cases passed"