Method test_nonzero_size

test/backend/test_ops.py:3360–3367

(self)

  @unittest.skipIf(COMPILE_ONLY, "test requires runtime")
  def test_nonzero_size(self):
    np.testing.assert_equal(Tensor([1, 0, 2, 0, 3]).nonzero(size=3).numpy(), [[0], [2], [4]])
    np.testing.assert_equal(Tensor([1, 0, 2, 0, 3]).nonzero(size=5, fill_value=-1).numpy(), [[0], [2], [4], [-1], [-1]])
    np.testing.assert_equal(Tensor([[1, 0], [0, 2]]).nonzero(size=2).numpy(), [[0, 0], [1, 1]])
    self.assertEqual(Tensor(5).nonzero(size=4).shape, (4, 0))
    np.testing.assert_equal(Tensor([], dtype=dtypes.int32).nonzero(size=3, fill_value=-1).numpy(), [[-1], [-1], [-1]])
    # fill_value must not promote dtype to float
    self.assertEqual(Tensor([1, 0]).nonzero(size=3, fill_value=-1.5).dtype, dtypes.default_int)

  def test_cast(self):
    helper_test_op([(3, 3)], lambda x: x.float())