def test_transpose_and_square_matrix_elements(self):
x = np.array([[1, 2], [3, 4]]).astype(float)
x = x.transpose()
expected = np.array(
[[1, 9],
[4, 16]])
actual = eigen.square(x)
np.testing.assert_array_equal(expected, actual)
def test_square_vector_elements(self):
x = np.array([1, 2, 3]).astype(float)
expected = np.array([[1], [4], [9]])
actual = pyspiel_eigen_test.square(x)
np.testing.assert_array_equal(expected, actual)
def test_transpose_then_slice_and_square_matrix_elements(self):
x = np.array([[1, 2], [3, 4]]).astype(float)
x = x.transpose()
expected = np.array([[9], [16]])
actual = pyspiel_eigen_test.square(x[0:, 1:])
np.testing.assert_array_equal(expected, actual)
def test_square_matrix_elements(self):
x = np.array([[1, 2], [3, 4]]).astype(float)
expected = np.array([[1, 2], [3, 4]])**2
actual = pyspiel_eigen_test.square(x)
np.testing.assert_array_equal(expected, actual)
