def test_array_mult(self): x = VectorVariable(3, 'x', label='dummy variable') x_0 = Monomial('x', idx=(0, ), shape=(3, ), label='dummy variable') x_1 = Monomial('x', idx=(1, ), shape=(3, ), label='dummy variable') x_2 = Monomial('x', idx=(2, ), shape=(3, ), label='dummy variable') p = x_0**2 + x_1**2 + x_2**2 self.assertEqual(x.dot(x), p) m = NomialArray([[x_0**2, x_0 * x_1, x_0 * x_2], [x_0 * x_1, x_1**2, x_1 * x_2], [x_0 * x_2, x_1 * x_2, x_2**2]]) self.assertEqual(x.outer(x), m)
def test_array_mult(self): x = VectorVariable(3, 'x', label='dummy variable') x_0 = Monomial('x', idx=(0,), shape=(3,), label='dummy variable') x_1 = Monomial('x', idx=(1,), shape=(3,), label='dummy variable') x_2 = Monomial('x', idx=(2,), shape=(3,), label='dummy variable') p = x_0**2 + x_1**2 + x_2**2 self.assertEqual(x.dot(x), p) m = NomialArray([[x_0**2, x_0*x_1, x_0*x_2], [x_0*x_1, x_1**2, x_1*x_2], [x_0*x_2, x_1*x_2, x_2**2]]) self.assertEqual(x.outer(x), m)
def test_outer(self): x = VectorVariable(3, 'x') y = VectorVariable(3, 'y') self.assertEqual(np.outer(x, y), x.outer(y)) self.assertEqual(np.outer(y, x), y.outer(x)) self.assertTrue(isinstance(x.outer(y), NomialArray))