Diffusion C-tracker (master) Browse

R3351/ctracker/modules/test_cross.py
a9fb993214f6master

test_cross.py
View Options

			import numpy as np
			from .cross import cross

			def test_cross_1():
			flux = np.ones((2, 2, 2, 2))
			sp, sn = cross(1, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 0.5
			assert sn == 1e20

			def test_cross_2():
			flux = np.ones((2, 2, 2, 2))
			sp, sn = cross(2, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 0.5
			assert sn == 1e20

			def test_cross_3():
			flux = np.ones((2, 2, 2, 2))
			sp, sn = cross(3, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 0.5
			assert sn == 1e20

			def test_cross_4():
			flux = -np.ones((2, 2, 2, 2))
			sp, sn = cross(1, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 1e20
			assert sn == 0.5

			def test_cross_5():
			flux = -np.ones((2, 2, 2, 2))
			sp, sn = cross(2, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 1e20
			assert sn == 0.5

			def test_cross_6():
			flux = -np.ones((2, 2, 2, 2))
			sp, sn = cross(3, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 1e20
			assert sn == 0.5

			def test_cross_7():
			flux = np.ones((2, 2, 2, 2))
			flux[..., 1] = -1.0
			sp, sn = cross(1, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 1e20
			assert sn == 1e20

			def test_cross_8():
			flux = np.ones((2, 2, 2, 2))
			flux[..., 1] = 0.0
			sp, sn = cross(1, 0, 0, 0, 1.0, 0.5, 0.5, 0.5, flux)
			assert sp == 1e20
			assert sn == 1e20

			def test_cross_9():
			flux = np.ones((2, 2, 2, 2))
			sp, sn = cross(1, 0, 0, 0, 1.0, 0.5, 0.5, 1.0 - 1e-6, flux)
			assert np.fabs(1 - sp/1e-6) < 1e-6
			assert sn == 1e20