diff options
| author | Stan Seibert <stan@mtrr.org> | 2011-11-17 14:02:40 -0500 |
|---|---|---|
| committer | tlatorre <tlatorre@uchicago.edu> | 2021-05-09 08:42:38 -0700 |
| commit | 7f2c55fcec9029fd79846f50e778876c6fd77f50 (patch) | |
| tree | 147b323d12bf3c0646311d738ae59807f2821aeb /test | |
| parent | 20d70f5064083925680abf6b93f4ff9442f4cb32 (diff) | |
| download | chroma-7f2c55fcec9029fd79846f50e778876c6fd77f50.tar.gz chroma-7f2c55fcec9029fd79846f50e778876c6fd77f50.tar.bz2 chroma-7f2c55fcec9029fd79846f50e778876c6fd77f50.zip | |
Replace all uses of numpy.count_nonzero() with chroma.tools.count_nonzero to remove requirement to use numpy >= 1.6
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_propagation.py | 3 |
1 files changed, 2 insertions, 1 deletions
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()) |