Skip to content

Commit 9ccf0cd

Browse files
committed
Add new tests
1 parent 7b69304 commit 9ccf0cd

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

tests/test_metrics.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def test_loss():
119119
Path(TEST_ASREVIEW_FILES, "sim_van_de_schoot_2017_stop_if_min.asreview")
120120
) as s:
121121
loss_value = loss(s)
122-
assert_almost_equal(loss_value, 0.011590940352087164, decimal=6)
122+
assert_almost_equal(loss_value, 0.011592855205548452, decimal=6)
123123

124124
def test_loss_value_function():
125125
labels = [1, 0]
@@ -132,11 +132,15 @@ def test_loss_value_function():
132132

133133
labels = [1, 1, 0, 0, 0]
134134
loss_value = _loss_value(labels)
135-
assert_almost_equal(loss_value, 0, decimal=6)
135+
assert_almost_equal(loss_value, 0, decimal=6), f"{loss_value} is {int(loss_value)}"
136136

137137
labels = [0, 0, 0, 1, 1]
138138
loss_value = _loss_value(labels)
139-
assert_almost_equal(loss_value, 1, decimal=6)
139+
assert_almost_equal(loss_value, 1, decimal=6)
140+
141+
labels = [1, 0, 1]
142+
loss_value = _loss_value(labels)
143+
assert_almost_equal(loss_value, 0.5, decimal=6)
140144

141145
import random
142146
for i in range(100):

0 commit comments

Comments
 (0)