@@ -4,7 +4,7 @@ defmodule Scholar.Preprocessing.CountVectorizer do
44 doctest CountVectorizer
55
66 describe "fit_transform" do
7- test "fit_transform test " do
7+ test "without padding " do
88 tesnsor = Nx . tensor ( [ [ 2 , 3 , 0 ] , [ 1 , 4 , 4 ] ] )
99
1010 counts =
@@ -17,7 +17,7 @@ defmodule Scholar.Preprocessing.CountVectorizer do
1717 assert counts == expected_counts
1818 end
1919
20- test "fit_transform test - tensor with padding" do
20+ test "with padding" do
2121 tensor = Nx . tensor ( [ [ 2 , 3 , 0 ] , [ 1 , 4 , - 1 ] ] )
2222
2323 counts =
@@ -30,12 +30,12 @@ defmodule Scholar.Preprocessing.CountVectorizer do
3030 end
3131
3232 describe "max_token_id" do
33- test "max_token_id test " do
33+ test "without padding " do
3434 tensor = Nx . tensor ( [ [ 2 , 3 , 0 ] , [ 1 , 4 , 4 ] ] )
3535 assert CountVectorizer . max_token_id ( tensor ) == 4
3636 end
3737
38- test "max_token_id tes - tensor with padding" do
38+ test "with padding" do
3939 tensor = Nx . tensor ( [ [ 2 , 3 , 0 ] , [ 1 , 4 , - 1 ] ] )
4040 assert CountVectorizer . max_token_id ( tensor ) == 4
4141 end
0 commit comments