Skip to content

Commit 0e81efa

Browse files
committed
Provide arguments in the test
1 parent 9e64145 commit 0e81efa

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/test_metrics.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,8 +220,15 @@ def test_source_target_len(self):
220220
X = np.array([[1, 2, 3], [4, 5, 6]])
221221
Y = np.array([[1, 2, 3]])
222222

223+
train_idx = [0]
224+
test_idx = [1]
225+
scaler = None
226+
estimator = None
227+
223228
with self.assertRaises(ValueError) as context:
224-
check_global_reconstruction_measures_input(X, Y)
229+
check_global_reconstruction_measures_input(
230+
X, Y, train_idx, test_idx, scaler, estimator
231+
)
225232

226233
expected_message = "First dimension of X (2) and Y (1) must match"
227234
self.assertEqual(str(context.exception), expected_message)

0 commit comments

Comments
 (0)