pandas.testing.assert_frame_equalの良い利用方法

前提・実現したいこと

前提

プログラムが組めない人向けに、csvデータの前処理ツールを作成しています。
csvをDataFrameにしてから様々な処理を行います。
csvのデータフォーマットは様々です。

実現したいこと

ソースコード改修によるバグを抑えるためにテストを行いたい。
テストは、DataFrame同士を比較しイコールかどうかで行いたい(pandas.testing.assert_frame_equal)
過去の生成したcsvに対してもテストを同時に行えるようにしたい。

発生している問題

下記のコードでは、うまくいったcsvを手動でTestCaseフォルダに移す必要がある
過去にBというパターンで成功していた場合、同時にパターンAとBのテストを行えない。

該当のソースコード

python
1## データの前処理 ##
2
3## DataFrameテスト
4# TestCaseフォルダには、成功したCSVを格納しておく（パターンAのテスト)
5testcasedf = pd.read_csv(os.getcwd() + r'/output_processingdata/' + 'output.csv')
6validationdf = pd.read_csv(os.getcwd() + r'/TestCase/' + 'validation.csv')
7pd.testing.assert_frame_equal(testcasedf , validationdf)

解決策が確認できるものがあれば教えてください。(ソース、URL、書籍等)

行動規範の内容に同意します

回答1件

ベストアンサー

下記のコードでは、うまくいったcsvを手動でTestCaseフォルダに移す必要がある

こちらに関しては,下記のようなイメージで実装できないでしょうか？
動作確認していないのでバグがあったらすいません。

try:
  pd.testing.assert_frame_equal(testcasedf , validationdf)
  validationdf.to_csv('os.getcwd() + r'/TestCase/')
except AssertionError as err:
  print('AssertionError :', err)

投稿2019/08/01 14:26

編集2019/08/01 14:28