def setUp(self) -> None: self.conn = psycopg2.connect(**TEST_DATABASE) with self.conn.cursor() as cursor: init_tables(cursor) fill_tables(cursor) self.conn.commit()
import psycopg2 from relational_database.config import DATABASE from relational_database.db_utils import init_tables, fill_tables if __name__ == "__main__": con = psycopg2.connect(**DATABASE) with con.cursor() as cursor: try: init_tables(cursor) fill_tables(cursor) con.commit() finally: if con: con.close()