def test_database_existence(self): db = Database(db = '') databases = list(db.execute('show databases;').Database) assert 'autonotrader' in databases, 'there is no autonotrader database' assert DB in databases, 'there is no test database'
from dotenv import load_dotenv from utils.database import Database import pandas as pd load_dotenv() db = Database() select_1 = db.select("""select * from users where id_user_pk < 4""") select_2 = db.select("""select * from users where id_user_pk < %(id)s""", {'id': 4}) select_3 = db.pd_select("""select * from users where id_user_pk < 4""") select_4 = db.pd_select("""select * from users where id_user_pk < %(id)s""", {'id': 4}) db.execute("""INSERT INTO users (name, age) VALUES ('abc', 12)""") db.execute("""INSERT INTO users (name, age) VALUES (%(name)s, %(age)s)""", { 'name': 'xsd', 'age': 45 }) values_list = [{'name': 'xsd', 'age': 45}, {'name': 'adsfs', 'age': 23}] df = pd.DataFrame([{'name': 'daaf', 'age': 22}, {'name': 'adf', 'age': 55}]) db.executemany("""INSERT INTO users (name, age) VALUES (%(name)s, %(age)s)""", values_list) db.pd_executemany( """INSERT INTO users (name, age) VALUES (%(name)s, %(age)s)""", df)