Create a Mock SQL DB in Python from CSV for unit testing Созд | Python/ django

Create a Mock SQL DB in Python from CSV for unit testing

Создание макета SQL-базы данных в Python из CSV для модульного тестирования.

pip install pandas
pip install sqlglot
pip install sqlalchemy
from sqlalchemy import create_engine, text
import sqlglot
import pandas as pd


def execute_sql_query(sql):
query_as_sqlite = sqlglot.transpile(sql, read="postgres", write="sqlite")[0]
mocked_db = create_engine('sqlite://')
pd.read_csv('data.csv').to_sql('table_name', con=mocked_db)
with mocked_db.connect() as connection:
result = connection.execute(text(query_as_sqlite))
return result

@pythonl