def test_noop(self): df = make_dataframe() df.values[[2, 5, 10], [2, 3, 1]] = np.nan y, X = df[df.columns[0]], df[df.columns[1:]] data, _ = sm_data.handle_missing(y, X, missing='none') y_exp, X_exp = df[df.columns[0]], df[df.columns[1:]] assert_frame_equal(data['exog'], X_exp) assert_series_equal(data['endog'], y_exp)
def test_array_pandas(self): df = make_dataframe() df.values[[2, 5, 10], [2, 3, 1]] = np.nan y, X = df[df.columns[0]].values, df[df.columns[1:]] data, _ = sm_data.handle_missing(y, X, missing='drop') df = df.dropna() y_exp, X_exp = df[df.columns[0]].values, df[df.columns[1:]] assert_frame_equal(data['exog'], X_exp) np.testing.assert_array_equal(data['endog'], y_exp)
def test_pandas_freq_decorator(): x = make_dataframe() # in x, get a function back that returns an x with the same columns func = pandas_wrapper(dummy_func) np.testing.assert_equal(func(x.values), x) func = pandas_wrapper(dummy_func_array) assert_frame_equal(func(x), x) expected = x.rename(columns=dict(zip('ABCD', 'EFGH'))) func = pandas_wrapper(dummy_func_array, names=list('EFGH')) assert_frame_equal(func(x), expected)