def test_array_set_ufunc_many_b(self):
a1 = np.array([3, 2, 1])
a2 = np.array([3, 2])
a3 = np.array([5, 3, 2, 1])
a4 = np.array([2])
post = array_set_ufunc_many((a1, a2, a3, a4), union=False)
self.assertEqual(post.tolist(), [2])
post = array_set_ufunc_many((a1, a2, a3, a4), union=True)
self.assertEqual(post.tolist(), [1, 2, 3, 5])
def test_array_set_ufunc_many_c(self):
a1 = np.array([[3, 2, 1], [1, 2, 3]])
a2 = np.array([[5, 2, 1], [1, 2, 3]])
a3 = np.array([[10, 20, 30], [1, 2, 3]])
post = array_set_ufunc_many((a1, a2, a3), union=False)
self.assertEqual(post.tolist(), [[1, 2, 3]])
post = array_set_ufunc_many((a1, a2, a3), union=True)
self.assertEqual(post.tolist(),
[[1, 2, 3], [3, 2, 1], [5, 2, 1], [10, 20, 30]])
def test_array_set_ufunc_many_a(self) -> None:
a1 = np.array([3, 2, 1])
a2 = np.array([3, 2, 1])
a3 = np.array([3, 2, 1])
a4 = np.array([3, 2, 1])
post = array_set_ufunc_many((a1, a2, a3, a4), union=False)
self.assertEqual(post.tolist(), [3, 2, 1])
post = array_set_ufunc_many((a1, a2, a3, a4), union=True)
self.assertEqual(post.tolist(), [3, 2, 1])
def test_array_set_ufunc_many_e(self):
a1 = np.array([3, 2, 1])
a2 = np.array([30, 20])
post = array_set_ufunc_many((a1, a2), union=False)
self.assertEqual(post.tolist(), [])
def test_array_set_ufunc_many_d(self):
a1 = np.array([3, 2, 1])
a2 = np.array([[5, 2, 1], [1, 2, 3]])
with self.assertRaises(Exception):
post = array_set_ufunc_many((a1, a2), union=False)
