def test_catch_different_lengths(self, data): with pytest.raises(ValueError): table1_generator( data, cols=['exp', 'mod', 'y', 'continuous'], variable_type=['category', 'category', 'continuous'], continuous_measure='A')
def test_wrong_continuous_measure_error(self, data): with pytest.raises(ValueError): table1_generator(data, cols=['exp', 'mod', 'y', 'continuous'], variable_type=['category', 'category', 'category', 'continuous'], continuous_measure='A') with pytest.raises(ValueError): table1_generator(data, cols=['mod', 'y', 'continuous'], variable_type=['category', 'category', 'continuous'], continuous_measure='A', strat_by='exp')
def test_stratified_median(self, data): t = table1_generator( data, cols=['mod', 'y', 'continuous'], variable_type=['category', 'category', 'continuous'], strat_by='exp') assert isinstance(t, type(pd.DataFrame()))
def test_unstratified_mean(self, data): t = table1_generator( data, cols=['exp', 'mod', 'y', 'continuous'], variable_type=['category', 'category', 'category', 'continuous'], continuous_measure='mean') assert isinstance(t, type(pd.DataFrame()))