def test_to_json():
try:
print("\n*********** Testing to JSON. *******************\n")
cat = CSVCatalog.CSVCatalog()
cat.drop_table("teams")
cds = []
cds.append(CSVCatalog.ColumnDefinition('teamID', 'text', True))
cds.append(CSVCatalog.ColumnDefinition('yearID', 'text', True))
cds.append(CSVCatalog.ColumnDefinition('W', column_type='number'))
tbl = CSVCatalog.TableDefinition(
"teams",
"/Users/donaldferguson/Dropbox/ColumbiaCourse/Courses/Fall2018/W4111/Homeworks/HW3/Data/core/Teams.csv",
column_definitions=cds)
r = json.dumps(tbl.to_json(), indent=2)
print("Teams definition = \n", r)
with open("unit_tests_catalog_json.txt", "w") as result_file:
result_file.write(r)
print("\n\n")
except Exception as e:
print(
"My implementation throws a custom exception. You can print any meaningful error you want."
)
print("Could not create table. Exception = ", e)
def test_create_table_3_fail():
"""
Creates a table that includes several column definitions. This test should fail because one of the defined
columns is not in the underlying CSV file.
:return:
"""
print_test_separator("Starting test_create_table_3_fail")
cleanup()
cat = CSVCatalog.CSVCatalog()
cds = []
cds.append(CSVCatalog.ColumnDefinition("playerID", "text", True))
cds.append(CSVCatalog.ColumnDefinition("nameLast", "text", True))
cds.append(CSVCatalog.ColumnDefinition("nameFirst", column_type="text"))
cds.append(CSVCatalog.ColumnDefinition("canary"))
try:
t = cat.create_table(
"people",
"/Users/donaldferguson/Dropbox/ColumbiaCourse/Courses/Fall2018/W4111-Projects/CSVDB/Data/core/People.csv",
cds)
print_test_separator("FAILURE test_create_table_3")
print("People table", json.dumps(t.describe_table(), indent=2))
except Exception as e:
print("Exception e = ", e)
print_test_separator("Complete test_create_table_3_fail successfully")
def test_create_table_5():
"""
Modifies a preexisting/precreated table definition.
:return:
"""
print_test_separator("Starting test_create_table_5")
# DO NOT CALL CLEANUP. Want to access preexisting table.
cat = CSVCatalog.CSVCatalog()
t = cat.get_table("batting")
print("Initial status of table = \n", json.dumps(t.describe_table(), indent=2))
t.add_column_definition(CSVCatalog.ColumnDefinition("HR", "number"))
t.add_column_definition(CSVCatalog.ColumnDefinition("G", "number"))
t.define_index("team_year_idx", "INDEX", ['teamID', 'yearID'])
print("Modified status of table = \n", json.dumps(t.describe_table(), indent=2))
print_test_separator("Success test_create_table_5")
def test_create_table_3():
"""
Creates a table that includes several column definitions.
:return:
"""
print_test_separator("Starting test_create_table_3")
cleanup()
cat = CSVCatalog.CSVCatalog()
cds = []
cds.append(CSVCatalog.ColumnDefinition("playerID", "text", True))
cds.append(CSVCatalog.ColumnDefinition("nameLast", "text", True))
cds.append(CSVCatalog.ColumnDefinition("nameFirst", column_type="text"))
t = cat.create_table(
"people",
"/Users/donaldferguson/Dropbox/ColumbiaCourse/Courses/Fall2018/W4111-Projects/CSVDB/Data/core/People.csv",
cds)
print("People table", json.dumps(t.describe_table(), indent=2))
print_test_separator("Complete test_create_table_3")
def test_create_table_5_prep():
"""
Creates a table that includes several column definitions and a primary key.
:return:
"""
print_test_separator("Starting test_create_table_5_prep")
cleanup()
cat = CSVCatalog.CSVCatalog()
cds = []
cds.append(CSVCatalog.ColumnDefinition("playerID", "text", True))
cds.append(CSVCatalog.ColumnDefinition("teamID", "text", True))
cds.append(
CSVCatalog.ColumnDefinition("yearID",
column_type="text",
not_null=True))
cds.append(
CSVCatalog.ColumnDefinition("stint",
column_type="number",
not_null=True))
cds.append(
CSVCatalog.ColumnDefinition("H", column_type="number", not_null=False))
cds.append(
CSVCatalog.ColumnDefinition("AB", column_type="number",
not_null=False))
t = cat.create_table(
"batting",
"/Users/donaldferguson/Dropbox/ColumbiaCourse/Courses/Fall2018/W4111-Projects/CSVDB/Data/core/Batting.csv",
cds)
t.define_primary_key(['playerID', 'teamID', 'yearID', 'stint'])
print("Batting table", json.dumps(t.describe_table(), indent=2))
print_test_separator("Completed test_create_table_5_prep")
def test_create_table_4_fail():
"""
Creates a table that includes several column definitions and a primary key.
The primary key references an undefined column, which is an error.
NOTE: You should check for other errors. You do not need to check in the CSV file for uniqueness but
should test other possible failures.
:return:
"""
print_test_separator("Starting test_create_table_4_fail")
cleanup()
cat = CSVCatalog.CSVCatalog()
cds = []
cds.append(CSVCatalog.ColumnDefinition("playerID", "text", True))
cds.append(CSVCatalog.ColumnDefinition("teamID", "text", True))
cds.append(CSVCatalog.ColumnDefinition("yearID", column_type="text", not_null=True))
cds.append(CSVCatalog.ColumnDefinition("stint", column_type="number", not_null=True))
cds.append(CSVCatalog.ColumnDefinition("H", column_type="number", not_null=False))
cds.append(CSVCatalog.ColumnDefinition("AB", column_type="number", not_null=False))
t = cat.create_table("batting",
"/Users/donaldferguson/Dropbox/ColumbiaCourse/Courses/Fall2018/W4111-Projects/CSVDB/Data/core/Batting.csv",
cds)
try:
t.define_primary_key(['playerID', 'teamID', 'yearID', 'HR'])
print("Batting table", json.dumps(t.describe_table(), indent=2))
print_test_separator("FAILURES test_create_table_4_fail")
except Exception as e:
print("Exception e = ", e)
print_test_separator("SUCCESS test_create_table_4_fail should fail.")