Python poissonの例

コード例 #1

ファイル: test_2D.py プロジェクト: zezhong-zhang/hyperspy

    def test_add_poisson_noise(self):
        s = self.signal
        kwargs = {}
        if s._lazy:
            data = s.data.compute()
            from dask.array.random import seed, poisson

            kwargs["chunks"] = s.data.chunks
        else:
            data = s.data.copy()
            from numpy.random import seed, poisson
        seed(1)

        s.add_poissonian_noise(keep_dtype=False)

        if s._lazy:
            s.compute()
        seed(1)
        np.testing.assert_array_almost_equal(s.data, poisson(lam=data,
                                                             **kwargs))
        s.change_dtype("float64")
        seed(1)

        s.add_poissonian_noise(keep_dtype=True)
        if s._lazy:
            s.compute()
        assert s.data.dtype == np.dtype("float64")

コード例 #2

ファイル: test_tools.py プロジェクト: woozey/hyperspy

 def test_add_poisson_noise(self):
     s = self.signal
     kwargs = {}
     if s._lazy:
         data = s.data.compute()
         from dask.array.random import seed, poisson
         kwargs["chunks"] = s.data.chunks
     else:
         data = s.data.copy()
         from numpy.random import seed, poisson
     seed(1)
     s.add_poissonian_noise(keep_dtype=False)
     if s._lazy:
         s.compute()
     seed(1)
     np.testing.assert_array_almost_equal(
         s.data, poisson(lam=data, **kwargs))
     s.change_dtype("float64")
     seed(1)
     s.add_poissonian_noise(keep_dtype=True)
     if s._lazy:
         s.compute()
     assert s.data.dtype == np.dtype("float64")