Skip to content

Commit

Permalink
Ensure test file follows convention
Browse files Browse the repository at this point in the history
  • Loading branch information
josevalim committed Jan 17, 2025
1 parent da18bb3 commit 48aa36e
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ defmodule Scholar.Preprocessing.CountVectorizer do
doctest CountVectorizer

describe "fit_transform" do
test "fit_transform test" do
test "without padding" do
tesnsor = Nx.tensor([[2, 3, 0], [1, 4, 4]])

counts =
Expand All @@ -17,7 +17,7 @@ defmodule Scholar.Preprocessing.CountVectorizer do
assert counts == expected_counts
end

test "fit_transform test - tensor with padding" do
test "with padding" do
tensor = Nx.tensor([[2, 3, 0], [1, 4, -1]])

counts =
Expand All @@ -30,12 +30,12 @@ defmodule Scholar.Preprocessing.CountVectorizer do
end

describe "max_token_id" do
test "max_token_id test" do
test "without padding" do
tensor = Nx.tensor([[2, 3, 0], [1, 4, 4]])
assert CountVectorizer.max_token_id(tensor) == 4
end

test "max_token_id tes - tensor with padding" do
test "with padding" do
tensor = Nx.tensor([[2, 3, 0], [1, 4, -1]])
assert CountVectorizer.max_token_id(tensor) == 4
end
Expand Down

0 comments on commit 48aa36e

Please sign in to comment.