pytest でフィクスチャを作成する
ユニットテストを作成する際には、テストを始める前に多少の準備が必要になることがあります。この準備をユニットテストの中で行うと、テストが読みにくくなったり、同じ処理を何度も繰り返したりすることになりかねません。幸い、pytest にはフィクスチャという、この問題を解決する機能があります。
この演習では、pandas が pd としてインポート済みで、以下の extract() 関数が使用できる状態になっています。
def extract(file_path):
return pd.read_csv(file_path)
この演習はコースの一部です
Python で学ぶ ETL と ELT
演習の手順
pytestライブラリをインポートします。raw_tax_dataという名前のpytestフィクスチャを作成します。raw_dataDataFrame を返します。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____