Home
last modified time | relevance | path

Searched refs:assert_array_equal (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/
H A Dtest_arraysetops.py27 from numpy.testing import assert_array_equal, assert_equal, assert_raises_regex
32 from torch._numpy.testing import assert_array_equal, assert_equal
46 assert_array_equal(c, ec)
54 assert_array_equal(c, ed)
55 assert_array_equal([], intersect1d([], []))
65 assert_array_equal(res, a)
67 assert_array_equal(res, [1, 2, 3])
75 assert_array_equal(c, ee)
76 assert_array_equal(a[i1], ee)
77 assert_array_equal(b[i2], ee)
[all …]
H A Dtest_twodim_base.py46 assert_array_equal,
71 assert_array_equal,
239 assert_array_equal(H.T, answer)
241 assert_array_equal(H.T, answer)
243 assert_array_equal(H, eye(10, 10))
244 assert_array_equal(xedges, np.linspace(0, 9, 11))
245 assert_array_equal(yedges, np.linspace(0, 9, 11))
262 assert_array_equal(xed, np.linspace(0, 6, 7))
263 assert_array_equal(yed, np.linspace(0, 5, 6))
275 assert_array_equal(H, 0)
[all …]
H A Dtest_function_base.py74 assert_array_equal,
106 assert_array_equal,
267 assert_array_equal(np.any(y1, axis=0), [1, 1, 0])
268 assert_array_equal(np.any(y1, axis=1), [0, 1, 1])
284 assert_array_equal(np.all(y1, axis=0), [0, 0, 1])
285 assert_array_equal(np.all(y1, axis=1), [0, 0, 1])
292 assert_array_equal(a, a_copy)
356 assert_array_equal(avg, expected_avg)
360 assert_array_equal(wavg, expected_wavg)
366 assert_array_equal(wavg, expected_wavg)
[all …]
H A Dtest_index_tricks.py30 assert_array_equal,
47 assert_array_equal,
252 assert_array_equal(c[0][0, :], -np.ones(10, "d"))
253 assert_array_equal(c[1][:, 0], -2 * np.ones(10, "d"))
312 assert_array_equal(grid128c_a, grid128c_b[0])
336 assert_array_equal(grid64_a, grid64_b)
341 assert_array_equal(grid64_a, grid64_b)
347 assert_array_equal(r_[1, 2, 3, 4, 5, 6], np.array([1, 2, 3, 4, 5, 6]))
350 assert_array_equal(c, [1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1])
374 assert_array_equal(d[:, :5], b)
[all …]
H A Dtest_shape_base_.py40 from numpy.testing import assert_, assert_array_equal, assert_equal
59 from torch._numpy.testing import assert_, assert_array_equal, assert_equal
169 assert_array_equal(apply_along_axis(len, 0, a), len(a) * np.ones(a.shape[1]))
173 assert_array_equal(apply_along_axis(len, 0, a), len(a) * np.ones(a.shape[1]))
177 assert_array_equal(
185 assert_array_equal(res, np.array([6, 6, 6]).view(cls))
196 assert_array_equal(res, np.array([6, 6, 6]).view(cls))
200 assert_array_equal(res, np.array([3, 3, 3, 3, 3, 3]).view(cls))
251 assert_array_equal(res[:, :, 0].mask, f1to2(a[:, 0]).mask)
252 assert_array_equal(res[:, :, 1].mask, f1to2(a[:, 1]).mask)
[all …]
H A Dtest_type_check.py30 from numpy.testing import assert_, assert_array_equal, assert_equal
44 from torch._numpy.testing import assert_, assert_array_equal, assert_equal
136 assert_array_equal(y, np.real(y))
140 assert_array_equal(y, out)
154 assert_array_equal(y.real, np.real(y))
158 assert_array_equal(y.real, out)
172 assert_array_equal(0, np.imag(y))
176 assert_array_equal(0, out)
190 assert_array_equal(y.imag, np.imag(y))
194 assert_array_equal(y.imag, out)
[all …]
H A Dtest_histograms.py30 assert_array_equal,
41 assert_array_equal,
56 assert_array_equal(a, 10)
61 assert_array_equal(
67 assert_array_equal(edges, [1, 2])
90 assert_array_equal(a, [1, 2, 3, 4])
170 assert_array_equal(hist, int_hist)
171 assert_array_equal(edges, int_edges)
191 assert_array_equal(wa, [4, 5, 0, 1])
244 assert_array_equal(a, np.array([0]))
[all …]
H A Dtest_arraypad.py17 from numpy.testing import assert_allclose, assert_array_equal
20 from torch._numpy.testing import assert_allclose, assert_array_equal
177 assert_array_equal(a, b)
331 assert_array_equal(a, b)
568 assert_array_equal(test, expected)
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/
H A Dtest_shape_base.py36 from numpy.testing import assert_, assert_array_equal, assert_equal
51 from torch._numpy.testing import assert_, assert_array_equal, assert_equal
66 assert_array_equal(res, desired)
73 assert_array_equal(res, desired)
80 assert_array_equal(res, desired)
89 assert_array_equal(res, desired)
105 assert_array_equal(res, desired)
112 assert_array_equal(res, desired)
119 assert_array_equal(res, desired)
128 assert_array_equal(res, desired)
[all …]
H A Dtest_numeric.py46 assert_array_equal,
58 assert_array_equal,
94 assert_array_equal(Ar, np.array([]))
107 assert_array_equal(Ar, np.zeros((2, 1), Ar.dtype))
455 assert_array_equal(e, ~d)
466 assert_array_equal(~self.t, self.f)
467 assert_array_equal(np.abs(~self.t), self.f)
468 assert_array_equal(np.abs(~self.f), self.t)
469 assert_array_equal(np.abs(self.f), self.f)
470 assert_array_equal(~np.abs(self.f), self.t)
[all …]
H A Dtest_dlpack.py25 from numpy.testing import assert_array_equal
28 from torch._numpy.testing import assert_array_equal
87 assert_array_equal(x, y)
93 assert_array_equal(y1, np.from_dlpack(y1))
96 assert_array_equal(y2, np.from_dlpack(y2))
99 assert_array_equal(y3, np.from_dlpack(y3))
102 assert_array_equal(y4, np.from_dlpack(y4))
105 assert_array_equal(y5, np.from_dlpack(y5))
141 assert_array_equal(x, y)
146 assert_array_equal(a, np.asarray(t))
[all …]
H A Dtest_multiarray.py48 assert_array_equal,
63 assert_array_equal,
457 assert_array_equal(x, y)
468 assert_array_equal(x["a"], [3.5, 3.5])
469 assert_array_equal(x["b"], [-2, -2])
532 assert_array_equal(e, [1, 3, 3])
539 assert_array_equal(e, [1, 4, 3])
541 assert_array_equal(d, [1, 4, 7])
548 assert_array_equal(e, [[1, 2, -7], [1, 2, 3]])
549 assert_array_equal(d, [[1, 3, 3], [1, 2, 3]])
[all …]
H A Dtest_indexing.py29 assert_array_equal,
38 assert_array_equal,
155 assert_array_equal(arr[index], arr[u_index])
157 assert_array_equal(arr, np.arange(10))
160 assert_array_equal(arr[index], arr[u_index])
163 assert_array_equal(arr, np.arange(5)[:, None].repeat(2, axis=1))
166 assert_array_equal(arr[u_index, u_index], arr[index, index])
312 assert_array_equal(a[i], a[i.copy("C")])
375 assert_array_equal(a[idx], idx)
383 assert_array_equal(a, res)
[all …]
H A Dtest_einsum.py14 assert_array_equal,
710 assert_array_equal(
714 assert_array_equal(np.einsum("ij,ij->j", p, q, optimize=True), [10.0] * 2)
720 assert_array_equal(np.einsum("i, i", x, y, optimize=False), 20.0)
721 assert_array_equal(np.einsum("i, i", x, y, optimize=True), 20.0)
727 assert_array_equal(
731 assert_array_equal(
739 assert_array_equal(
742 assert_array_equal(
745 assert_array_equal(
[all …]
H A Dtest_scalarmath.py650 assert_array_equal(arr_like * np.float32(3.0), np.full(3, 3.0))
651 assert_array_equal(np.float32(3.0) * arr_like, np.full(3, 3.0))
652 assert_array_equal(arr_like * np.int_(3), np.full(3, 3))
653 assert_array_equal(np.int_(3) * arr_like, np.full(3, 3))
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dremote_cluster_test.py175 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
180 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
192 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
197 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
217 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
235 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
241 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
261 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
288 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
293 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
[all …]
H A Dremote_execution_test.py100 np.testing.assert_array_equal(x1.numpy(), x2.numpy())
110 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
122 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
133 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
158 np.testing.assert_array_equal([[4.0]], loss.numpy())
170 np.testing.assert_array_equal([[9.0]], loss.numpy())
171 np.testing.assert_array_equal([[6.0]], grad.numpy())
188 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
203 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
215 np.testing.assert_array_equal([[2, 2], [2, 2]], y.numpy())
[all …]
/aosp_15_r20/external/pytorch/test/torch_np/
H A Dtest_reductions.py27 assert_array_equal,
36 assert_array_equal,
363 assert_array_equal(func(a, axis=()), func(np.expand_dims(a, axis=0), axis=0))
385 assert_array_equal(with_keepdims, expanded)
394 assert_array_equal(with_keepdims, expanded)
409 assert_array_equal(result, result_with_out)
421 assert_array_equal(result, result_with_out)
453 assert_array_equal(result, result_with_out)
514 assert_array_equal(func(a, axis=None), func(a.ravel(), axis=0))
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/
H A Dtf_transpose_test.py82 np.testing.assert_array_equal(res, np.transpose(arg0, (0, 2, 1)))
107 np.testing.assert_array_equal(res, np.transpose(arg0, (2, 0, 1)))
132 np.testing.assert_array_equal(res, np.transpose(arg0, (2, 1, 0)))
157 np.testing.assert_array_equal(res, np.transpose(arg0, (1, 2, 0)))
182 np.testing.assert_array_equal(res, np.transpose(arg0, (1, 0, 2)))
/aosp_15_r20/external/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/
H A Dannotations_unittest.py121 np.testing.assert_array_equal(e.GetLevel(), data['level'])
123 np.testing.assert_array_equal(
126 np.testing.assert_array_equal(
129 np.testing.assert_array_equal(
132 np.testing.assert_array_equal(
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_dtensor.py4 from numpy.testing import assert_array_equal
851 assert_array_equal(expected_pad_sizes, pad_sizes)
858 assert_array_equal(expected_is_tensor_empty, is_tensor_empty)
864 assert_array_equal(expected_pad_sizes, pad_sizes)
882 assert_array_equal(expected_is_tensor_empty, is_tensor_empty)
/aosp_15_r20/external/pytorch/torch/_numpy/testing/
H A D__init__.py9 assert_array_equal,
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/linalg/
H A Dtest_linalg.py55 assert_array_equal,
88 assert_array_equal,
515 assert_array_equal(result, expected)
531 assert_array_equal(result, expected)
550 assert_array_equal(result, expected)
556 assert_array_equal(result, expected)
1928 assert_array_equal(res, routine(n_arr))
1930 assert_array_equal(res, routine(sw_arr))
/aosp_15_r20/external/armnn/python/pyarmnn/test/
H A Dtest_tensor.py124 np.testing.assert_array_equal(copied_tensor.get_memory_area(), initial_mem_copy)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client_test.py760 np.testing.assert_array_equal(y.to_py(), z.to_py())
2330 np.testing.assert_array_equal(
2370 np.testing.assert_array_equal(x, buffer.to_py())
2371 np.testing.assert_array_equal(x, y.to_py())
2372 np.testing.assert_array_equal(x, z.to_py())
2398 np.testing.assert_array_equal(x, y)
2418 np.testing.assert_array_equal(x, y)
2826 np.testing.assert_array_equal(out.to_py(), args[0] * args[1])

12