我正在使用psycopg2-binary連接到資料庫。這是我撰寫的類，用于在給定查詢的情況下從資料庫中查詢資料并作為資料框回傳。我一直堅持如何對這個函式進行單元測驗，而不必訪問真正的資料庫。
代碼 ：
import pandas as pd
class HitDB(object):
def __init__(self, db_con):
self.conn = db_con
def load_data(self, query):
df = pd.read_sql(
sql = query,
con = self.conn
)
return df
uj5u.com熱心網友回復：
由于該函式所做的只是將其查詢傳遞self.conn給另一個函式，因此模擬pd.read_sql并檢查它是否接收到預期的引數。
這建議擺脫此功能并直接呼叫pd.read_sql。
如果它對資料庫做了更重要的事情，請運行本地 PostgreSQL 服務器（本地或使用 Docker 之類的東西），加載模式，并使用工廠根據需要使用測驗資料填充它。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/529357.html