def test_method_insert_7(): ad = AstroData(TESTFILE) xname = "TEST" xver = 99 del header1['EXTNAME'] ad.insert(1, header=header1, data=data1, extname=xname, extver=xver, auto_number=True) assert ad[1].header.get("EXTNAME") == xname assert ad[1].header.get("EXTVER") == xver
def runinsert(index=None, f1=None, f2=None, auto=False): ad = AstroData(f1) md = AstroData(f2) pstr = "\n\n >>>>>>> AD <<<<<<<<\n" pstr += str(ad.infostr()) pstr += "\n\n >>>>>>> AD APPEND <<<<<<<<\n" pstr += str(md.infostr()) ad.insert(index=index, moredata=md, auto_number=auto) pstr +="\n\n >>>>>>> NEW AD <<<<<<<<\n" pstr += str(ad.infostr()) print(pstr) return ad
def test_method_insert_6(): ad = AstroData(TESTFILE) del header1['EXTNAME'] ad.insert(1, header=header1, data=data1, extname="TEST", auto_number=True) assert ad[1].header.get("EXTNAME") == "TEST"
def test_method_insert_5(): ad = AstroData(TESTFILE) del header1['EXTNAME'] with pytest.raises(KeyError): ad.insert(1, header=header1, data=data1, auto_number=True)
def test_method_insert_4(): ad = AstroData(TESTFILE) initial_len = len(ad) ad.insert(1, moredata=pfob, auto_number=True) assert len(ad) == initial_len + len(pfob) - 1
def test_method_insert_3(): ad = AstroData(TESTFILE) initial_len = len(ad) ad.insert(1, moredata=hdu1, extname='TEST', auto_number=True) with pytest.raises(KeyError): assert ad[1].header['TEST']
def test_method_insert_1(): ad = AstroData(TESTFILE) ad2 = AstroData(TESTFILE2) initial_len = len(ad) ad.insert(1, moredata=ad2, auto_number=True) assert len(ad) == initial_len + 1
def test_method_insert_9(): ad = AstroData(TESTFILE) initial_len = len(ad) header1['EXTNAME'] = "TEST" with pytest.raises(AstroDataError): ad.insert(1, header=header1, auto_number=True)
def test_method_insert_8(): ad = AstroData(TESTFILE) initial_len = len(ad) header1['EXTNAME'] = 'TEST' ad.insert(1, header=header1, data=data1, auto_number=True) assert ad[1].header['EXTNAME'] == 'TEST'
def test_method_insert_10(): ad = AstroData(TESTFILE) initial_len = len(ad) del header1['EXTNAME'] with pytest.raises(AstroDataError): ad.insert(1, data=data1, auto_number=True)