質問編集履歴
1
暫定対応の追記
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -103,3 +103,53 @@
|
|
|
103
103
|
============================================= 1 failed in 2.50s =============================================
|
|
104
104
|
|
|
105
105
|
```
|
|
106
|
+
|
|
107
|
+
|
|
108
|
+
|
|
109
|
+
---
|
|
110
|
+
|
|
111
|
+
■追記
|
|
112
|
+
|
|
113
|
+
|
|
114
|
+
|
|
115
|
+
DataFrame同士で比較するために、比較関数を作成する必要があるとのことで、
|
|
116
|
+
|
|
117
|
+
当初の質問のような、DataFrame同士の比較という形ではなく、
|
|
118
|
+
|
|
119
|
+
DataFrameの列単位での比較という形でテストを実装しました。
|
|
120
|
+
|
|
121
|
+
|
|
122
|
+
|
|
123
|
+
* テストケース(test_model.py)
|
|
124
|
+
|
|
125
|
+
```
|
|
126
|
+
|
|
127
|
+
import pandas as pd
|
|
128
|
+
|
|
129
|
+
import model
|
|
130
|
+
|
|
131
|
+
|
|
132
|
+
|
|
133
|
+
def test_01():
|
|
134
|
+
|
|
135
|
+
# 入力値
|
|
136
|
+
|
|
137
|
+
dict_input = {'name': ['userA'], 'height': [1.11]}
|
|
138
|
+
|
|
139
|
+
df_input: pd.DataFrame = pd.DataFrame(data = dict_input)
|
|
140
|
+
|
|
141
|
+
# 期待値
|
|
142
|
+
|
|
143
|
+
dict_expected = {'name': ['userA'], 'height': [1.11], 'height^2': [1.2321]}
|
|
144
|
+
|
|
145
|
+
df_expected: pd.DataFrame = pd.DataFrame(data = dict_expected)
|
|
146
|
+
|
|
147
|
+
# 出力値
|
|
148
|
+
|
|
149
|
+
df_output: pd.DataFrame = model.calc_square(df_input)
|
|
150
|
+
|
|
151
|
+
# 比較
|
|
152
|
+
|
|
153
|
+
assert df_expected.equals(df_output.round(4) # round()にて小数点以下4桁で丸め
|
|
154
|
+
|
|
155
|
+
```
|