def test_outcome_fraction_should_compute_percentage_of_positive_outcome(self): df = pd.DataFrame(data=np.random.rand(12, 2), columns=["var1", "var2"]) df["Outcome"] = [random.sample(range(2), 1)[0] for i in range(12)] value = len(df[df["Outcome"] == 1]) / len(df) result = utils.outcome_fraction_(df) self.assertEqual(value, result)
def test_outcome_fraction_should_compute_percentage_of_positive_outcome_with_custom_name( self): df = pd.DataFrame(data=np.random.rand(12, 2), columns=['var1', 'var2']) df['result'] = [random.sample(range(2), 1)[0] for i in range(12)] value = len(df[df['result'] == 1]) / len(df) result = utils.outcome_fraction_(df, col_outcome='result') self.assertEqual(value, result)