def _eval_cartesian_multi_helper(self, **kwargs):
nodes = np.asfortranarray([[2.0, 0.0, 3.0], [3.0, 2.0, 7.5]])
triangle = self._make_one(nodes, 1, copy=False)
param_vals = np.asfortranarray([[1.0, 0.0]])
patch = unittest.mock.patch(
"bezier._triangle_helpers.evaluate_cartesian_multi",
return_value=unittest.mock.sentinel.evaluated,
)
with patch as mocked:
result = triangle.evaluate_cartesian_multi(param_vals, **kwargs)
self.assertEqual(result, unittest.mock.sentinel.evaluated)
mocked.assert_called_once_with(nodes, 1, param_vals, 2)
def test_elevate_linear(self):
nodes = np.asfortranarray([[0.0, 2.0, -1.0], [0.0, 1.0, 2.0]])
triangle = self._make_one(nodes, 1)
elevated = triangle.elevate()
expected = np.asfortranarray([[0.0, 1.0, 2.0, -0.5, 0.5, -1.0],
[0.0, 0.5, 1.0, 1.0, 1.5, 2.0]])
self.assertEqual(triangle._degree, 1)
self.assertEqual(elevated._degree, 2)
self.assertEqual(elevated._nodes, expected)
main_vals = triangle.evaluate_cartesian_multi(self.REF_TRIANGLE3)
sub_vals = elevated.evaluate_cartesian_multi(self.REF_TRIANGLE3)
self.assertEqual(main_vals, sub_vals)
def test_elevate_quadratic(self):
klass = self._get_target_class()
nodes = np.asfortranarray([[0.0, 6.0, 9.0, 0.0, 6.0, -3.0]])
triangle = klass.from_nodes(nodes)
elevated = triangle.elevate()
expected = np.asfortranarray(
[[0.0, 4.0, 7.0, 9.0, 0.0, 4.0, 7.0, -1.0, 3.0, -3.0]])
self.assertEqual(triangle._degree, 2)
self.assertEqual(elevated._degree, 3)
self.assertEqual(elevated._nodes, expected)
main_vals = triangle.evaluate_cartesian_multi(self.REF_TRIANGLE3)
sub_vals = elevated.evaluate_cartesian_multi(self.REF_TRIANGLE3)
self.assertEqual(main_vals, sub_vals)
def test_evaluate_cartesian_multi_wrong_dimension(self):
triangle = self._make_one(self.ZEROS, 1)
param_vals_1d = np.zeros((4, ), order="F")
with self.assertRaises(ValueError):
triangle.evaluate_cartesian_multi(param_vals_1d)