質問編集履歴
1
暫定対応の追記
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -50,4 +50,29 @@
|
|
|
50
50
|
=============================================== short test summary info ===============================================
|
|
51
51
|
FAILED tests/src/test_model.py::test_0１ - assert False
|
|
52
52
|
============================================= 1 failed in 2.50s =============================================
|
|
53
|
+
```
|
|
54
|
+
|
|
55
|
+
---
|
|
56
|
+
■追記
|
|
57
|
+
|
|
58
|
+
DataFrame同士で比較するために、比較関数を作成する必要があるとのことで、
|
|
59
|
+
当初の質問のような、DataFrame同士の比較という形ではなく、
|
|
60
|
+
DataFrameの列単位での比較という形でテストを実装しました。
|
|
61
|
+
|
|
62
|
+
* テストケース(test_model.py)
|
|
63
|
+
```
|
|
64
|
+
import pandas as pd
|
|
65
|
+
import model
|
|
66
|
+
|
|
67
|
+
def test_01():
|
|
68
|
+
# 入力値
|
|
69
|
+
dict_input = {'name': ['userA'], 'height': [1.11]}
|
|
70
|
+
df_input: pd.DataFrame = pd.DataFrame(data = dict_input)
|
|
71
|
+
# 期待値
|
|
72
|
+
dict_expected = {'name': ['userA'], 'height': [1.11], 'height^2': [1.2321]}
|
|
73
|
+
df_expected: pd.DataFrame = pd.DataFrame(data = dict_expected)
|
|
74
|
+
# 出力値
|
|
75
|
+
df_output: pd.DataFrame = model.calc_square(df_input)
|
|
76
|
+
# 比較
|
|
77
|
+
assert df_expected.equals(df_output.round(4) # round()にて小数点以下4桁で丸め
|
|
53
78
|
```
|