Changeset da8925 in git for machine_learning/common/keyword_vector.py

Timestamp:

Jul 30, 2019, 6:27:35 PM (5 years ago)

Author:

Murray Heymann <heymann.murray@…>

Branches:

(u'spielwiese', 'fe61d9c35bf7c61f2b6cbf1b56e25e2f08d536cc')

Children:

50e8722ce7219d67ee649c39300d30d2e9e90edd

Parents:

f598830d8683f1182349ba48230ab28a08198004

git-author:

Murray Heymann <heymann.murray@gmail.com>2019-07-30 18:27:35+02:00

git-committer:

Murray Heymann <heymann.murray@gmail.com>2019-07-30 18:28:16+02:00

Message:

start migrating to unittest framework

File:

• machine_learning/common/keyword_vector.py (modified) (1 diff)

machine_learning/common/keyword_vector.py

@@ -123 +123 @@
         vec = vec / norm
     return vec
-
-def test_read_dictionary():
-    """
-    Create test for read_dictionary function
-    """
-    print("\033[1;32;40mTesting read_dictionary function:\033[1;37;40m")
-
-    print("Non-existant file")
-    correct = False
-    try:
-        read_dictionary("asdfasdf")
-    except FileNotFoundError:
-        print("correctly caught non-existant file")
-        print("\033[1;32;40mpass\033[1;37;40m")
-        correct = True
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-    print("Reading default file")
-    correct = True
-    try:
-        read_dictionary()
-    except FileNotFoundError:
-        print("Default file for dictionary missing")
-        print("\033[1;31;40mfail\033[1;37;40m")
-        correct = False
-    if correct:
-        print("\033[1;32;40mpass\033[1;37;40m")
-    print()
-    print()
-
-def test_count_occurances():
-    """
-    Create test for count_occurances function
-    """
-    print("\033[1;32;40mTesting count_occurances function:\033[1;37;40m")
-    dic = read_dictionary()
-    correct = False
-    try:
-        vec = count_occurances("asdfasdf", dic)
-    except FileNotFoundError:
-        correct = True
-        print("Correctly raised FileNotFoundError")
-        print("\033[1;32;40mpass\033[1;37;40m")
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-    print("Count occurances with None dictionary:")
-    correct = False
-    try:
-        count_occurances("../Singular/table.h", None)
-    except AssertionError:
-        print("Correctly caught AssertionError")
-        print("\033[1;32;40mpass\033[1;37;40m")
-        correct = True
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-
-    print("Count occurances with empty dictionary:")
-    correct = False
-    try:
-        count_occurances("../Singular/table.h", np.array([]))
-    except AssertionError:
-        print("Correctly caught AssertionError")
-        print("\033[1;32;40mpass\033[1;37;40m")
-        correct = True
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-
-    print("vector of ../Singular/table.h")
-    vec = count_occurances("../Singular/table.h", dic)
-    print(vec)
-    print()
-    print()
-
-def test_create_vector_dictionary():
-    """
-    Create test for create_vector_dictionary function
-    """
-    print("\033[1;32;40mTesting create_vector_dictionary " \
-            "function:\033[1;37;40m")
-    read_dictionary()
-
-    print("Create Vector Dictionary with None as dictionary:")
-    correct = False
-    try:
-        create_vector_dictionary(None)
-    except AssertionError:
-        correct = True
-        print("\033[1;32;40mpass\033[1;37;40m")
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-    print("Create Vector Dictionary with empty dictionary:")
-    correct = False
-    try:
-        create_vector_dictionary(np.array([]))
-    except AssertionError:
-        correct = True
-        print("\033[1;32;40mpass\033[1;37;40m")
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-    print()
-    print()
-
-def test_vector_distance():
-    """
-    Create test for vector_distance function
-    """
-    print("\033[1;32;40mTesting vector_distance function:\033[1;37;40m")
-
-    vector1 = np.array([3, 4])
-    vector1 = normalise_vector(vector1)
-    vector2 = np.array([4, 3])
-    vector2 = normalise_vector(vector2)
-
-    print("Distance of vectors of different dimensions:")
-    correct = False
-    try:
-        vector_distance(np.array([1, 2, 3]), vector1)
-    except AssertionError:
-        correct = True
-        print("\033[1;32;40mpass\033[1;37;40m")
-    if not correct:
-        print("\033[1;31;40mfail\033[1;37;40m")
-
-
-    print("Distance same vector: " + str(vector_distance(vector1, vector1)))
-    assert vector_distance(vector1, vector1) == 0, \
-            "distance to same vectorshould be 0"
-    print("\033[1;32;40mpass\033[1;37;40m")
-
-    print("Distance different vector: " + str(vector_distance(vector1, vector2)))
-    assert vector_distance(vector1, vector2) > 0, \
-            "Distance between nonequal vectors should be strictly positive"
-    print("\033[1;32;40mpass\033[1;37;40m")
-    print()
-    print()
-
-def test_normalise_vector():
-    """
-    Create test for normalise_vector function
-    """
-    print("\033[1;32;40mTesting normalise_vector function:\033[1;37;40m")
-    testvector = np.array([3, 4])
-    testvector = normalise_vector(testvector)
-    assert np.linalg.norm(testvector) == 1, \
-            "Normalised vector should have norm of 1"
-    print("\033[1;32;40mpass\033[1;37;40m")
-    print("normalised vector: " + str(testvector))
-    print("\033[1;32;40mpass\033[1;37;40m")
-
-    print("Attempt to normalise the zero vector")
-    print(normalise_vector(np.array([0, 0, 0, 0, 0])))
-    print("\033[1;32;40mpass\033[1;37;40m")
-
-    print("Attempt to normalise list")
-    print(normalise_vector([3, 4, 0, 0, 0]))
-    print("\033[1;32;40mpass\033[1;37;40m")
-
-    print("Attempt to normalise empty vector")
-    print(normalise_vector(np.array([])))
-    print("\033[1;32;40mpass\033[1;37;40m")
-
-    print("Attempt to normalise None")
-    print(normalise_vector(None))
-    print("\033[1;32;40mpass\033[1;37;40m")
-    print()
-    print()
-
-
-def main():
-    """
-    Run some basic tests
-    """
-    test_normalise_vector()
-
-    test_vector_distance()
-
-    test_read_dictionary()
-
-    test_count_occurances()
-
-    test_create_vector_dictionary()
-
-if __name__ == '__main__':
-    main()