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", "data/Teams.csv", column_definitions=cds,
cnx=cat.cnx) # maybe this is inproper?
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)
print("\n*********** Testing to JSON. *******************\n")
def test_table_def_to_json():
print("\n*********** Testing table def to JSON. *******************\n")
cds = list()
cds.append(CSVCatalog.ColumnDefinition('teamID', 'text', True))
cds.append(CSVCatalog.ColumnDefinition('yearID', 'text', True))
cds.append(CSVCatalog.ColumnDefinition('AB', 'number'))
tbl = CSVCatalog.TableDefinition(
"batting",
"../data/Batting.csv",
column_definitions=cds)
print(tbl.to_json())
def test2():
a = CSVCatalog.TableDefinition()
a.define_primary_key(['playerID', 'teamID', 'yearID', 'HR'])
a.define_index("team_year_idx", ['teamID', 'yearID'], "INDEX")
def test1():
a = CSVCatalog.TableDefinition()
a.define_primary_key(['playerID', 'teamID', 'yearID', 'HR'])
