如何在Python中为数据库创建存根?

时间:2019-04-09 15:13:20

标签: database python-3.x pandas mocking stub

我希望能够创建一个数据库存根(或模拟对象),以便测试一个查询数据库的函数,而不是使用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)

0 个答案:

没有答案