remove NeuriteType.all from NEURITES (#931)

BlueBrain · Jun 15, 2021 · 76ce6b3 · 76ce6b3
1 parent 8d2fc96
commit 76ce6b3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -10,6 +10,7 @@ Version 2.3.0
 - Remove the cli command ``neurom features`` that listed all possible features. Instead a proper
   documentation is provided on that topic. See :func:`neurom.features.get`.
 - Make ``neurom.features.neuronfunc.sholl_crossings`` private.
+- Remove ``NeuriteType.all`` from ``NEURITES``
 
 Version 2.2.1
 -------------

diff --git a/neurom/core/types.py b/neurom/core/types.py
@@ -62,8 +62,7 @@ class NeuriteType(IntEnum):
 
 
 #: Collection of all neurite types
-NEURITES = (NeuriteType.all,
-            NeuriteType.axon,
+NEURITES = (NeuriteType.axon,
             NeuriteType.apical_dendrite,
             NeuriteType.basal_dendrite)
 

diff --git a/tests/core/test_types.py b/tests/core/test_types.py
@@ -54,11 +54,9 @@ def test_tree_type_checker():
     assert tree_filter(mock_tree)
 
     tree_filter = tree_type_checker(*NEURITES)
-    assert tree_filter('fake_tree')
     assert tree_filter(mock_tree)
 
     tree_filter = tree_type_checker(NEURITES)
-    assert tree_filter('fake_tree')
     assert tree_filter(mock_tree)