我將測試my_function()，它返回一個(gè)帶有“utc_timestamp”列的 pandas DataFrame。此列包含當(dāng)前時(shí)間戳。def my_function(): df = pd.DataFrame(data=[[1, pd.datetime.utcnow()]], columns=['value', 'utc_timestamp'])如何定義單元測試來測試my_function()的預(yù)期輸出。顯然下面的方法是行不通的，因?yàn)闇y試函數(shù)中的當(dāng)前時(shí)間戳不等于創(chuàng)建數(shù)據(jù)幀時(shí)的時(shí)間戳。class TestCase(unittest.TestCase): df_out = my_function() df_expected = pd.DataFrame(data=[[1, pd.datetime.utcnow()]], columns=['value', 'utc_timestamp']) pd.testing.assert_frame_equal(df_out, df_expected)>> AssertionError: DataFrame.iloc[:, 0] (column name="utc_timestamp") are different哪一個(gè)是最專業(yè)、最pythonic和最pandas-way的？
1 回答
慕尼黑8549860
TA貢獻(xiàn)1818條經(jīng)驗(yàn) 獲得超11個(gè)贊
我不確定你是否找到了單元測試的方法，但這里有一種單元測試你的 DataFrame 的方法......
您必須修補(bǔ)pd.datetime.utcnow函數(shù)并將其設(shè)置為返回靜態(tài)日期。然后，您可以使用 assert 語句來檢查兩個(gè) DataFrame 是否相等。
import unittest
import pandas as pd
from unittest.mock import patch
from pandas.util.testing import assert_frame_equal
with patch('pandas.datetime.utcnow', return_value="2020-08-16 20:36:06.578174"):
actual = my_function()
expected = pd.DataFrame(data=[[1, "2020-08-16 20:36:06.578174"]], columns=['value', 'utc_timestamp'])
assert_frame_equal(actual, expected)
添加回答
舉報(bào)
0/150
提交
取消