示例#1
0
 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')
示例#2
0
 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')
示例#3
0
 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()))
示例#4
0
 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()))