Dataframe null check doesn't provide df name if error occurs about pandera HOT 6 CLOSED

<ipython-input-29-dc85ea979a2b> in <module>
----> 1 schema.validate(df) <<< this dataframe

In [1]: import pandas as pd
   ...: import numpy as np
   ...: from pandera import Column, DataFrameSchema, PandasDtype, SeriesSchema, Check, Int
   ...:
   ...: schema = DataFrameSchema(
   ...:     {
   ...:         "a": Column(PandasDtype.Int, Check(lambda x: x > 0))
   ...:     })
   ...:
   ...: df = pd.DataFrame({
   ...:     "a": [1, 2, np.nan]
   ...: })
   ...:
   ...: schema.validate(df)
---------------------------------------------------------------------------
SchemaError                               Traceback (most recent call last)
<ipython-input-1-f4098e24342e> in <module>
     12 })
     13
---> 14 schema.validate(df)

~/git/pandera/pandera/pandera.py in validate(self, dataframe)
    180             return self.schema.validate(dataframe)
    181         except SchemaError as e:
--> 182             raise SchemaError(str(e)) from None
    183
    184

SchemaError: non-nullable series 'a' contains null values: {2: nan}

SchemaError                               Traceback (most recent call last)
<ipython-input-29-dc85ea979a2b> in <module>
----> 1 schema.validate(df)

/panderadev/pandera/pandera.py in validate(self, dataframe)
    176         if not isinstance(dataframe, pd.DataFrame):
    177             raise TypeError("expected dataframe, got %s" % type(dataframe))
--> 178         return self.schema.validate(dataframe)
    179 
    180 

/mycondaenv/lib/python3.6/site-packages/schema.py in validate(self, data)
    393                 return s.validate(data)
    394             except SchemaError as x:
--> 395                 raise SchemaError([None] + x.autos, [e] + x.errors)
    396             except BaseException as x:
    397                 message = "%r.validate(%r) raised %r" % (s, data, x)

SchemaError: expected series 'a' to have type int64, got float64 and non-nullable series contains null values: {2: nan}

---------------------------------------------------------------------------
SchemaError                               Traceback (most recent call last)
<ipython-input-1-f4098e24342e> in <module>
     12 })
     13
---> 14 schema.validate(df)

~/git/pandera/pandera/pandera.py in validate(self, dataframe)
    172         if self.index is not None:
    173             schema_elements += [self.index]
--> 174         assert all(s(dataframe) for s in schema_elements)
    175         if self.transformer is not None:
    176             dataframe = self.transformer(dataframe)

~/git/pandera/pandera/pandera.py in <genexpr>(.0)
    172         if self.index is not None:
    173             schema_elements += [self.index]
--> 174         assert all(s(dataframe) for s in schema_elements)
    175         if self.transformer is not None:
    176             dataframe = self.transformer(dataframe)

~/git/pandera/pandera/pandera.py in __call__(self, df)
    332             raise RuntimeError(
    333                 "need to `set_name` of column before calling it.")
--> 334         return super(Column, self).__call__(df[self._name])
    335
    336     def __repr__(self):

~/git/pandera/pandera/pandera.py in __call__(self, series)
    222                 raise SchemaError(
    223                     "non-nullable series contains null values: %s" %
--> 224                     series[nulls].head(N_FAILURE_CASES).to_dict()
    225                 )
    226

SchemaError: non-nullable series contains null values: {2: nan}

import pandas as pd
import numpy as np
from pandera import Column, DataFrameSchema, PandasDtype, SeriesSchema, Check, Int

# Define input and output schemas
schema_in = DataFrameSchema(
    {
        "a": Column(PandasDtype.Int, Check(lambda x: x > 0))
    })
schema_out = DataFrameSchema(
    {
        "a": Column(PandasDtype.Float, Check(lambda x: x > 0))
    })

# This dataframe will fail schema_input, but pass schema_out
df = pd.DataFrame({
    "a": [1, 2, np.nan]
})

@check_input(schema_in, 0)
@check_output(schema_out)
def some_function(df):
    return df

function_return = some_function(df)

---------------------------------------------------------------------------
SchemaError                               Traceback (most recent call last)
<ipython-input-24-de765c22ff69> in <module>
----> 1 function_return = some_function(df)

/panderadev/pandera/pandera.py in _wrapper(fn, instance, args, kwargs)
    376         args = list(args)
    377         if isinstance(obj_getter, int):
--> 378             args[obj_getter] = schema.validate(args[obj_getter])
    379         elif isinstance(obj_getter, str):
    380             if obj_getter in kwargs:

/panderadev/pandera/pandera.py in validate(self, dataframe)
    171         if self.index is not None:
    172             schema_elements += [self.index]
--> 173         assert all(s(dataframe) for s in schema_elements)
    174         if self.transformer is not None:
    175             dataframe = self.transformer(dataframe)

/panderadev/pandera/pandera.py in <genexpr>(.0)
    171         if self.index is not None:
    172             schema_elements += [self.index]
--> 173         assert all(s(dataframe) for s in schema_elements)
    174         if self.transformer is not None:
    175             dataframe = self.transformer(dataframe)

/panderadev/pandera/pandera.py in __call__(self, df)
    339             raise RuntimeError(
    340                 "need to `set_name` of column before calling it.")
--> 341         return super(Column, self).__call__(df[self._name])
    342 
    343     def __repr__(self):

/panderadev/pandera/pandera.py in __call__(self, series)
    225                         "non-nullable series contains null values: %s" %
    226                         (series.name, self._pandas_dtype.value, series.dtype,
--> 227                          series[nulls].head(N_FAILURE_CASES).to_dict()))
    228                 else:
    229                     raise SchemaError(

SchemaError: expected series 'a' to have type int64, got float64 and non-nullable series contains null values: {2: nan}