result = CasJobs.writeFitsFileFromQuery(fileName=CasJobs_TestFitsFile, queryString=CasJobs_TestQuery, context="MyDB") print(result) # In[ ]: #delete local FITS file just created: os.remove(CasJobs_TestFitsFile) # In[ ]: #get a Pandas dataframe containing the results of a query df = CasJobs.getPandasDataFrameFromQuery(queryString=CasJobs_TestQuery, context=CasJobs_TestDatabase) print(df) # In[ ]: # get numpy array containing the results of a query array = CasJobs.getNumpyArrayFromQuery(queryString=CasJobs_TestQuery, context=CasJobs_TestDatabase) print(array) # In[ ]: #uploads a Pandas dataframe into a Database table
def test_CasJobs_getPandasDataFrameFromQuery(self):
#CasJobs.getPandasDataFrameFromQuery
df = CasJobs.getPandasDataFrameFromQuery(queryString=CasJobs_TestQuery,
context=CasJobs_TestDatabase)
self.assertEqual(df.to_csv(index=False), CasJobs_TestTableCSV)
def get_query_result(self, tablename):
query = 'select * from MyDB.{}'.format(tablename)
df = CasJobs.getPandasDataFrameFromQuery(queryString=query,
context="MyDB")
df['distance'] = df['distance'] * u.arcmin
return df
