def test_invalid_data_frame_size(self): df_empty = pd.DataFrame({'A': []}) with self.assertRaises(AssertionError) as ctx: down_sample(df_empty, self.B, 100, 10) actual = str(ctx.exception) expected = 'Size of the input table is 0' self.assertEqual(actual, expected)
def test_down_sample_seed(self): C, D = down_sample(self.A, self.B, 100, 10, seed=0, show_progress=False) E, F = down_sample(self.A, self.B, 100, 10, seed=0, show_progress=False) self.assertEqual(D.equals(F), True) self.assertEqual(C.equals(E), True)
def test_down_sample_njobs_all(self): C, D = down_sample(self.A, self.B, 100, 10, seed=0, n_jobs=-1, show_progress=False)
def test_down_sample_njobs_fixed(self): C, D = down_sample(self.A, self.B, 100, 10, seed=0, n_jobs=2, show_progress=False) assert (len(C) > 0) assert (len(D) > 0)
def test_down_sample_invalid_df_sz0B(self): C, D = down_sample(self.A, pd.DataFrame(), 100, 10)
def test_down_sample_invalid_df_sz0A(self): C, D = down_sample(pd.DataFrame(), self.B, 100, 10)
def test_down_sample_table_invalid_dfB(self): C, D = down_sample(self.A, None, 100, 10)
def test_down_sample_table_invalid_dfA(self): C, D = down_sample(None, self.B, 100, 10)
def test_down_sample_table_valid_3(self): C, D = down_sample(self.A, self.B, 100, 10) self.assertNotEqual(len(C), 0)
def test_down_sample_table_valid_2(self): C, D = down_sample(self.A, self.B, len(self.B) + 1, 10) self.assertEqual(len(D), len(self.B))
def test_down_sample_invalid_rem_puncs(self): C, D = down_sample(self.A, self.B, 100, 10, rem_puncs="False")
def test_down_sample_invalid_rem_stop_words(self): C, D = down_sample(self.A, self.B, 100, 10, rem_stop_words="False")
def test_down_sample_invalid_njobs(self): C, D = down_sample(self.A, self.B, 100, 10, n_jobs="10")
def test_down_sample_invalid_seed(self): C, D = down_sample(self.A, self.B, 100, 10, seed="test")
def test_down_sample_invalid_param_size(self): C, D = down_sample(self.A, self.B, 0, 10)
def test_down_sample_invalid_param_y(self): C, D = down_sample(self.A, self.B, 100, 0)
def test_down_sample_table_valid_1(self): C, D = down_sample(self.A, self.B, 100, 10) self.assertEqual(len(D), 100)