def test_table_file_operations():
    table = Table("person_data.dat")
    table.set_attributes(["name", "age", "weight"])
    table.set_key("name")
    
    try:
        table.populate_initial_data("person_data_four_col.txt", "++++", "\n")
    except:
        a = 0 # This code does nothing, but is necessary for the except block
    else:
        print("populate_initial_data() must raise an exception when input file has incorrect format.")
    
    table.populate_initial_data("person_data_three_col.txt", "++++", "\n")
    if table.get_size() != 12:
        print("Table size not correct after populating initial data.")
    if table.sequential_search("first") != ["Abe Simons", "48", "176"]:
        print("First row of Table after populating initial data is incorrect.")
    if table.sequential_search("last") != ["Jules Rory", "25", "167"]:
        print("Last row of Table after populating initial data is incorrect.")
        
    table = Table("inverted_index_table.dat")
    table.set_type("inverted_index")
    table.load_one_document("story.txt", "101")
    if table.get_size() != 347:
        print("Table size incorrect after loading one document.")
    if table.sequential_search("first") != ["the", "101"]:
        print("First row of Table is incorrect after loading one document.")
    if table.key_search("darigan") != ["darigan", "101"]:
        print("Table is missing key(s) after loading one document.")
def test_table_data_and_search():
    table = Table("data_table.dat")
    table.set_attributes(["name", "age", "weight"])
    table.set_key("name")
    
    table.append(["Abe Simons", "48", "176"])
    table.append(["Billy Kidd", "17", "145"])
    table.append(["Emily Smith", "22", "138"])
    table.append(["Carla Cree", "34", "215"])
    table.append(["David Johnson", "71", "165"])
    table.append(["Katherine Diaz", "32", "144"])
    table.append(["Frank Small", "29", "129"])
    table.append(["George Sorenson", "16", "188"])
    table.append(["Lenny Peterson", "19", "201"])
    table.append(["Heidi Thomson", "55", "133"])
    table.append(["Iggy Bat", "61", "153"])
    table.append(["Jules Rory", "25", "167"])
    
    sequential_search_correct = True   
    if table.sequential_search("first") != ["Abe Simons", "48", "176"]:
        sequential_search_correct = False
    if table.sequential_search("last") != ["Jules Rory", "25", "167"]:
        sequential_search_correct = False
    if table.sequential_search("prev") != ["Iggy Bat", "61", "153"]:
        sequential_search_correct = False
    if table.sequential_search("next") != ["Jules Rory", "25", "167"]:
        sequential_search_correct = False
    if not sequential_search_correct:
        print("Sequential search does not work correctly.")
        
    if table.key_search("Nobody") != None:
        print("Key search must return None when key does not exist in Table.")
    if table.key_search("George Sorenson") != ["George Sorenson", "16", "188"]:
        print("Key search must return row when key does exist in Table.")
    if table.sequential_search("prev") != ["Frank Small", "29", "129"]:
        print("Current pointer not set correctly after key search.")