예제 #1
0
 def func(a):
     h1 = create_fitsheader(a.shape[::-1],
                            dtype=float if a.ndim > 0 else np.int32)
     h2 = create_fitsheader_for(a if a.ndim > 0 else None)
     assert h1 == h2
     assert h1['NAXIS'] == a.ndim
     for i, s in enumerate(a.shape[::-1]):
         assert h1['NAXIS'+str(i+1)] == s
예제 #2
0
def test_create_fitsheader2():
    header = create_fitsheader_for(None)
    assert header['NAXIS'] == 0