Source code for validphys.tests.test_tableloader

import pandas as pd
import numpy as np

from validphys.loader import FallbackLoader as Loader
from validphys import tableloader


def test_min_combination():
    nan = np.nan
    dfdicts = [
            {('NNPDF31_nlo_as_0130_uncorr__combined',
              'chi2'): {('NMC', 'NMC', 204, 394): nan, ('NMC',
               'NMC',
               204,
               400): 679.78465060721771, ('NMC', 'NMCPD', 121, 394): nan, ('NMC',
               'NMCPD',
               121,
               400): 226.73557951005108, ('NMC', 'Total', 325, 394): nan, ('NMC',
               'Total',
               325,
               400): 906.52023011726862, ('SLAC', 'SLACD', 34, 394): nan, ('SLAC',
               'SLACD',
               34,
               400): 55.48265988454024, ('SLAC', 'SLACP', 33, 394): nan, ('SLAC',
               'SLACP',
               33,
               400): 56.765548408772041, ('SLAC', 'Total', 67, 394): nan, ('SLAC',
               'Total',
               67,
               400): 110.50940462648715}},

          {('NNPDF31_nlo_as_0130_uncorr__combined', 'chi2'):
              {('NMC', 'NMC', 204, 394): 572.89990139102679, ('NMC','NMC', 204,
           400): 643.10896957499369, ('NMC',
           'NMCPD',
           121,
           394): 228.66185589823439, ('NMC', 'NMCPD', 121, 400): 232.87411189255329, ('NMC',
           'Total',
           325,
           394): 801.56175728926098, ('NMC',
           'Total',
           325,
           400): 875.98308146754709, ('SLAC', 'SLACD', 34, 394): 62.071957514383364, ('SLAC',
           'SLACD',
           34,
           400): 44.278821773436725, ('SLAC',
           'SLACP',
           33,
           394): 70.465557968855123, ('SLAC', 'SLACP', 33, 400): 57.213453384186352, ('SLAC',
           'Total',
           67,
           394): 126.25943979466223, ('SLAC', 'Total', 67, 400): 101.69776217874313}},
    ]
    dfs = [pd.DataFrame.from_dict(df) for df in dfdicts]
    res = tableloader.combine_pseudorreplica_tables(dfs, ['NNPDF31_nlo_as_0130_uncorr__combined'],  )
    assert pd.isnull(res.loc[pd.IndexSlice[:,:,:,394],:]).all().all()
    assert (res.loc[pd.IndexSlice[:,:,:,400],:] == dfs[1].loc[pd.IndexSlice[:,:,:,400],:]).all().all()
    res2 = tableloader.combine_pseudorreplica_tables(
            dfs, ['NNPDF31_nlo_as_0130_uncorr__combined'], min_points_required=1)
    assert not pd.isnull(res2.loc[pd.IndexSlice[:,:,:,394],:]).all().all()
    assert (res2.loc[pd.IndexSlice[:,:,:,400],:] == dfs[1].loc[pd.IndexSlice[:,:,:,400],:]).all().all()


def test_extrasum_slice():
    l = Loader()
    f =  l.check_vp_output_file('ljzWOixPQfmq5dA1-EUocg==/tables/fits_chi2_table.csv')
    d, l = tableloader.load_adapted_fits_chi2_table(f)
    components = ['LHCb Total', 'ATLASTTBARTOT', 'LHCBZ940PB']
    sliced = tableloader.get_extrasum_slice(d, components)
    slicel = tableloader.get_extrasum_slice(d, components)
    assert sliced.shape == (3,1)
    assert sliced.loc[('LHCb', 'Total'),:].shape == (1,)
    assert slicel.size == 3