From 7f2c55fcec9029fd79846f50e778876c6fd77f50 Mon Sep 17 00:00:00 2001 From: Stan Seibert Date: Thu, 17 Nov 2011 14:02:40 -0500 Subject: Replace all uses of numpy.count_nonzero() with chroma.tools.count_nonzero to remove requirement to use numpy >= 1.6 --- test/test_propagation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'test') diff --git a/test/test_propagation.py b/test/test_propagation.py index 34e64f5..e1b52ae 100644 --- a/test/test_propagation.py +++ b/test/test_propagation.py @@ -5,6 +5,7 @@ from chroma.geometry import Solid, Geometry, vacuum from chroma.make import box from chroma.sim import Simulation from chroma.event import Photons +from chroma.tools import count_nonzero class TestPropagation(unittest.TestCase): def testAbort(self): @@ -51,6 +52,6 @@ class TestPropagation(unittest.TestCase): max_steps=10).next().photons_end aborted = (photons_end.flags & (1 << 31)) > 0 print 'aborted photons: %1.1f' % \ - (float(np.count_nonzero(aborted)) / nphotons) + (float(count_nonzero(aborted)) / nphotons) self.assertFalse(aborted.any()) -- cgit