def test_generate_tpds(self):
table_generator = TableGenerator("tpcds", 0.001,
self.generating_connector)
# Check that lineitem table exists in TableGenerator
item_table = None
for table in table_generator.tables:
if table.name == "item":
item_table = table
break
self.assertIsNotNone(item_table)
# Check that i_item_sk column exists in TableGenerator and Table object
i_item_sk = Column("i_item_sk")
item_table.add_column(i_item_sk)
self.assertIn(i_item_sk, table_generator.columns)
self.assertIn(i_item_sk, table.columns)
database_connect = PostgresDatabaseConnector(
table_generator.database_name(), autocommit=True)
tpcds_tables = [
"call_center",
"catalog_page",
"catalog_returns",
"catalog_sales",
"customer",
"customer_address",
"customer_demographics",
"date_dim",
"household_demographics",
"income_band",
"inventory",
"item",
"promotion",
"reason",
"ship_mode",
"store",
"store_returns",
"store_sales",
"time_dim",
"warehouse",
"web_page",
"web_returns",
"web_sales",
"web_site",
]
for tpcds_table in tpcds_tables:
self.assertTrue(database_connect.table_exists(tpcds_table))
self.generating_connector.close()
database_connect.close()
def test_generate_tpch(self):
table_generator = TableGenerator("tpch", 0.001,
self.generating_connector)
# Check that lineitem table exists in TableGenerator
lineitem_table = None
for table in table_generator.tables:
if table.name == "lineitem":
lineitem_table = table
break
self.assertIsNotNone(lineitem_table)
# Check that l_receiptdate column exists in TableGenerator and Table object
l_receiptdate = Column("l_receiptdate")
lineitem_table.add_column(l_receiptdate)
self.assertIn(l_receiptdate, table_generator.columns)
self.assertIn(l_receiptdate, table.columns)
database_connect = PostgresDatabaseConnector(
table_generator.database_name(), autocommit=True)
tpch_tables = [
"customer",
"lineitem",
"nation",
"orders",
"part",
"partsupp",
"region",
"supplier",
]
for tpch_table in tpch_tables:
self.assertTrue(database_connect.table_exists(tpch_table))
self.generating_connector.close()
database_connect.close()