diff --git a/tests/test_cube3d.py b/tests/test_cube3d.py index 655157274d738ee70175068f1ac30c1da1ff2883..4d56a593d1818299b8b43d8042e763871e25311b 100644 --- a/tests/test_cube3d.py +++ b/tests/test_cube3d.py @@ -65,12 +65,15 @@ class test_cube3d(unittest.TestCase): u.savefits('result.fits') - plt.figure(figsize=(16,4)) - plt.plot(u.wave,np.log10(u.flux[50,50,:]),color='blue',label=r'$v=%4.2f$'%(velmap.map[50,50])) - plt.plot(u.wave,np.log10(u.flux[5,95,:]),color='red',label=r'$v=%4.2f$'%(velmap.map[5,95])) - plt.xlim(3500,9500) - plt.legend(frameon=False) - plt.savefig('./image/test_cube3d_spec.png') + if np.sum(u.flux[50,50,:]) != np.sum(u.flux[5,95,:]): + print("No Problem!") + + #plt.figure(figsize=(16,4)) + #plt.plot(u.wave,np.log10(u.flux[50,50,:]),color='blue',label=r'$v=%4.2f$'%(velmap.map[50,50])) + #plt.plot(u.wave,np.log10(u.flux[5,95,:]),color='red',label=r'$v=%4.2f$'%(velmap.map[5,95])) + #plt.xlim(3500,9500) + #plt.legend(frameon=False) + #plt.savefig('./image/test_cube3d_spec.png') print("-----------------------------------------------------------------") print(">>> Unit test for **Spec Cube** Modelling has been passed!")