Skip to content

Commit

Permalink
chore: uncomment asserts
Browse files Browse the repository at this point in the history
  • Loading branch information
polvalente committed Aug 7, 2023
1 parent 8f76a35 commit d60bed2
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions nx/test/nx_test.exs
Original file line number Diff line number Diff line change
Expand Up @@ -2524,14 +2524,14 @@ defmodule NxTest do

describe "load_numpy/1" do
test "loads array" do
# assert Nx.load_numpy!(File.read!("test/fixtures/numpy/no_dims_int64.npy")) ==
# Nx.tensor(123, type: {:s, 64})
assert Nx.load_numpy!(File.read!("test/fixtures/numpy/no_dims_int64.npy")) ==
Nx.tensor(123, type: {:s, 64})

# assert Nx.load_numpy!(File.read!("test/fixtures/numpy/1d_int64.npy")) ==
# Nx.tensor([1, 2, 3, 4], type: {:s, 64})
assert Nx.load_numpy!(File.read!("test/fixtures/numpy/1d_int64.npy")) ==
Nx.tensor([1, 2, 3, 4], type: {:s, 64})

# assert Nx.load_numpy!(File.read!("test/fixtures/numpy/2d_float32.npy")) ==
# Nx.tensor([[1, 2], [3, 4], [5, 6]], type: {:f, 32})
assert Nx.load_numpy!(File.read!("test/fixtures/numpy/2d_float32.npy")) ==
Nx.tensor([[1, 2], [3, 4], [5, 6]], type: {:f, 32})

assert Nx.load_numpy!(File.read!("test/fixtures/numpy/1d_uint8.npy")) == Nx.tensor([1, 2, 3], type: {:u, 8})
end
Expand Down

0 comments on commit d60bed2

Please sign in to comment.