Skip to content

Commit 26c0999

Browse files
committed
Updated style [skip ci]
1 parent 9bc2b1b commit 26c0999

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

test/booster_test.rb

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,11 @@ def test_predict_hash
3737
pred = booster.predict({"x3" => 9.0, "x2" => 7.2, "x1" => 1.2, "x0" => 3.7})
3838
assert_in_delta 0.9823112229173586, pred
3939

40-
pred = booster.predict(
41-
[
40+
pred =
41+
booster.predict([
4242
{"x3" => 9.0, "x2" => 7.2, "x1" => 1.2, "x0" => 3.7},
43-
{"x3" => 0.0, "x2" => 7.9, "x1" => 0.5, "x0" => 7.5},
44-
]
45-
)
43+
{"x3" => 0.0, "x2" => 7.9, "x1" => 0.5, "x0" => 7.5}
44+
])
4645
assert_elements_in_delta [0.9823112229173586, 0.9583143724610858], pred.first(2)
4746

4847
assert_raises(KeyError) do
@@ -54,7 +53,7 @@ def test_predict_daru
5453
x_test =
5554
Daru::DataFrame.new([
5655
{"x3" => 9.0, "x2" => 7.2, "x1" => 1.2, "x0" => 3.7},
57-
{"x3" => 0.0, "x2" => 7.9, "x1" => 0.5, "x0" => 7.5},
56+
{"x3" => 0.0, "x2" => 7.9, "x1" => 0.5, "x0" => 7.5}
5857
])
5958
pred = booster.predict(x_test)
6059
assert_elements_in_delta [0.9823112229173586, 0.9583143724610858], pred.first(2)
@@ -65,7 +64,7 @@ def test_predict_rover
6564
x_test =
6665
Rover::DataFrame.new([
6766
{"x3" => 9.0, "x2" => 7.2, "x1" => 1.2, "x0" => 3.7},
68-
{"x3" => 0.0, "x2" => 7.9, "x1" => 0.5, "x0" => 7.5},
67+
{"x3" => 0.0, "x2" => 7.9, "x1" => 0.5, "x0" => 7.5}
6968
])
7069
pred = booster.predict(x_test)
7170
assert_elements_in_delta [0.9823112229173586, 0.9583143724610858], pred.first(2)

0 commit comments

Comments
 (0)