From 35e5fd5c03847b103528ed439236543651dd11bd Mon Sep 17 00:00:00 2001 From: Michael Deistler Date: Mon, 6 Nov 2023 11:45:57 +0100 Subject: [PATCH] Fix tests --- setup.py | 1 + tests/test_swc.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 665d5028..a9acd35c 100644 --- a/setup.py +++ b/setup.py @@ -4,6 +4,7 @@ "jax[cpu]", "numpy", "pandas", + "matplotlib", ] diff --git a/tests/test_swc.py b/tests/test_swc.py index e4f9d236..eb3d8adf 100644 --- a/tests/test_swc.py +++ b/tests/test_swc.py @@ -20,7 +20,7 @@ def test_swc_reader_lengths(): fname = "morph.swc" - _, pathlengths, _, _ = nx.utils.read_swc(fname, max_branch_len=2000.0) + _, pathlengths, _, _ = nx.utils.swc.swc_to_neurax(fname, max_branch_len=2000.0) pathlengths = np.asarray(pathlengths)[1:] for sec in h.allsec(): @@ -58,7 +58,7 @@ def test_swc_radius(): # Can not use full morphology because of branch sorting. fname = "morph_250.swc" - _, pathlen, radius_fns, _ = nx.utils.read_swc( + _, pathlen, radius_fns, _ = nx.utils.swc.swc_to_neurax( fname, max_branch_len=2000.0, sort=False ) neurax_diams = []