1. 学ぶ
  2. /
  3. コース
  4. /
  5. Python で学ぶ ETL と ELT

Connected

演習

pytestでフィクスチャを作成する

ユニットテストを作成するとき、テストを始める前に少し準備が必要になることがあります。これを各ユニットテストの中で行うと、テストが読みにくくなり、同じ処理を何度も繰り返すことにもなりかねません。幸い、pytest にはフィクスチャという仕組みがあり、これらの問題を解決できます。

この演習では、pandas は pd としてインポート済みで、下記の extract() 関数を利用できます。

def extract(file_path):
    return pd.read_csv(file_path)

指示

100 XP
  • pytest ライブラリをインポートします。
  • raw_tax_data という名前の pytest フィクスチャを作成します。
  • raw_data DataFrame を返します。