我希望能够创建一个数据库存根(或模拟对象),以便测试一个查询数据库的函数,而不是使用Pandas返回数据帧的函数。当前,我无法正确创建正确模拟的数据库,因此无法创建数据框。我不确定我要去哪里。
def p2ctt_data_frame():
conn = pyodbc.connect(
r'Driver={Microsoft Access Driver (*.mdb, *.accdb)};'
r'DBQ=My\Path\To\Actual\Database\Access Database.accdb;'
)
query = 'select * from P2CTT_2016_Plus0HHs'
df = pd.read_sql(query, conn)
return df
class TestMockDatabase(unittest.TestCase):
@mock.patch('directory1.script1.pyodbc.connect')
def test_mock_database(self, mock_con):
mock_con.return_value.execute.return_value.fetchall.side_effect = [('1', '2', '3'), (1, 2, 3)]
expected = pd.DataFrame({
'1': [
1
],
'2': [
1
],
'3': [
3
]
})
result = p2ctt_data_frame()
tm.assert_frame_equal(result, expected)
我希望结果与预期的数据帧匹配,但事实并非如此。当前,我遇到以下错误:
AssertionError:DataFrame不同
DataFrame形状不匹配 [左] :( 0、0) [右] :( 1、3)