diff --git a/elastica/_elastica_numba/_joint.py b/elastica/_elastica_numba/_joint.py index 4235f974..22c5b2d5 100644 --- a/elastica/_elastica_numba/_joint.py +++ b/elastica/_elastica_numba/_joint.py @@ -186,7 +186,7 @@ def _calculate_contact_forces( @numba.njit(cache=True) def _aabbs_not_intersecting(aabb_one, aabb_two): - """ Returns true if not intersecting else false""" + """Returns true if not intersecting else false""" if (aabb_one[0, 1] < aabb_two[0, 0]) | (aabb_one[0, 0] > aabb_two[0, 1]): return 1 if (aabb_one[1, 1] < aabb_two[1, 0]) | (aabb_one[1, 0] > aabb_two[1, 1]): diff --git a/elastica/_elastica_numba/_rotations.py b/elastica/_elastica_numba/_rotations.py index 49c209c2..ca795507 100644 --- a/elastica/_elastica_numba/_rotations.py +++ b/elastica/_elastica_numba/_rotations.py @@ -46,6 +46,7 @@ def _get_rotation_matrix(scale: float, axis_collection): return rot_mat +@njit(cache=True) def _rotate(director_collection, scale: float, axis_collection): """ Does alibi rotations diff --git a/examples/Visualization/_povmacros.py b/examples/Visualization/_povmacros.py index 351e62ae..330bd4a0 100644 --- a/examples/Visualization/_povmacros.py +++ b/examples/Visualization/_povmacros.py @@ -189,7 +189,7 @@ def __init__(self, pre_scripts="", post_scripts=""): self._light_assign = defaultdict(list) def add_camera(self, name, **kwargs): - """ Add camera (viewpoint) """ + """Add camera (viewpoint)""" self.cameras.append(self.Camera(name=name, **kwargs)) def add_light(self, camera_id=-1, **kwargs): diff --git a/tests/test_elastica_numba/test_rod_nb.py b/tests/test_elastica_numba/test_rod_nb.py index 1e6ce078..567ab3a9 100644 --- a/tests/test_elastica_numba/test_rod_nb.py +++ b/tests/test_elastica_numba/test_rod_nb.py @@ -26,7 +26,7 @@ def __init__(self): # https://docs.pytest.org/en/latest/fixture.html @pytest.fixture(scope="module", params=[15, 31]) def load_data_for_bootstrapping_state(request): - """ Yield states for bootstrapping """ + """Yield states for bootstrapping""" n_elem = request.param n_nodes = n_elem + 1 dim = 3 diff --git a/tests/test_elastica_numba/test_timestepper_nb.py b/tests/test_elastica_numba/test_timestepper_nb.py index 0c9976aa..f394a8f1 100644 --- a/tests/test_elastica_numba/test_timestepper_nb.py +++ b/tests/test_elastica_numba/test_timestepper_nb.py @@ -37,7 +37,7 @@ class TestExtendStepperInterface: - """ TODO add documentation """ + """TODO add documentation""" class MockSymplecticStepper: Tag = SymplecticStepperTag() diff --git a/tests/test_elastica_numpy/test_rod_np.py b/tests/test_elastica_numpy/test_rod_np.py index ae9e0b3b..16004690 100644 --- a/tests/test_elastica_numpy/test_rod_np.py +++ b/tests/test_elastica_numpy/test_rod_np.py @@ -26,7 +26,7 @@ def __init__(self): # https://docs.pytest.org/en/latest/fixture.html @pytest.fixture(scope="module", params=[15, 31]) def load_data_for_bootstrapping_state(request): - """ Yield states for bootstrapping """ + """Yield states for bootstrapping""" n_elem = request.param n_nodes = n_elem + 1 dim = 3 diff --git a/tests/test_elastica_numpy/test_timestepper_np.py b/tests/test_elastica_numpy/test_timestepper_np.py index 597ee8f3..e3f5bf5b 100644 --- a/tests/test_elastica_numpy/test_timestepper_np.py +++ b/tests/test_elastica_numpy/test_timestepper_np.py @@ -37,7 +37,7 @@ class TestExtendStepperInterface: - """ TODO add documentation """ + """TODO add documentation""" class MockSymplecticStepper: Tag = SymplecticStepperTag()