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)
