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 ____