astrolabsoftware / fink-science-portal

Fink Science Portal
https://fink-portal.org
Apache License 2.0
7 stars 4 forks source link

Test SSOFT schema programatically #613

Closed JulienPeloton closed 3 months ago

JulienPeloton commented 5 months ago

Here is a meaningful test:

import io
import requests
import pandas as pd

r = requests.post(
  'https://fink-portal.org/api/v1/ssoft',
  json={
    'output-format': 'parquet'
  }
)

# Format output in a DataFrame
pdf = pd.read_parquet(io.BytesIO(r.content))

# get the schema
r = requests.get('https://fink-portal.org/api/v1/ssoft?schema')
schema = r.json()

# check columns
assert set(schema['args'].keys()) == set(pdf.columns)