Skip to content

Commit dc60426

Browse files
committed
fix issue with test assertions
1 parent b33640f commit dc60426

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

test/runtests.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,24 +12,24 @@ using TextAnalysis
1212

1313
# train transformer
1414
tfidf_transformer = MLJText.TfidfTransformer()
15-
test = machine(tfidf_transformer, ngram_vec)
16-
MLJBase.fit!(test)
15+
test_machine = machine(tfidf_transformer, ngram_vec)
16+
MLJBase.fit!(test_machine)
1717

1818
# test
1919
test_doc = ngrams(NGramDocument("Another sentence ok"))
20-
transform(test, [test_doc])
20+
test1 = transform(test_machine, [test_doc])
2121
@test sum(test1, dims=2)[1] == 0.0
2222
@test size(test1) == (1, 11)
2323

2424
test_doc2 = ngrams(NGramDocument("Listen Sam, today is not the day."))
25-
transform(test, [test_doc2])
25+
test2 = transform(test_machine, [test_doc2])
2626
@test sum(test2, dims=2)[1] > 0.0
2727
@test size(test2) == (1, 11)
2828

2929
test_doc3 = ngrams.(
3030
Corpus(NGramDocument("Another sentence ok"), NGramDocument("Listen Sam, today is not the day."))
3131
)
32-
transform(test, test_doc3)
32+
test3 = transform(test_machine, test_doc3)
3333
@test sum(test3, dims=2)[1] == 0.0
3434
@test sum(test3, dims=2)[2] > 0.0
3535
@test size(test3) == (2, 11)

0 commit comments

Comments
 (0)