Ejemplo n.º 1
0
def test_delete_na1():
	testing_df = df.copy()
	result_df = df.copy()
	clean.deleteRowsWithNA(testing_df, 0)
	result_df.dropna(subset=['col1'], inplace=True)
	result_df.reset_index(drop=True, inplace=True)
	assert (((testing_df.fillna(0) == result_df.fillna(0)).all()).all()) == True
Ejemplo n.º 2
0
def test_delete_na3():
	testing_df = df.copy()
	with pytest.raises(Exception):
		clean.deleteRowsWithNA(testing_df, 16) 
Ejemplo n.º 3
0
def test_delete_na2():
	testing_df = df.copy()
	result_df = df.copy()
	clean.deleteRowsWithNA(testing_df, 2)
	assert (((testing_df.fillna(0) == result_df.fillna(0)).all()).all()) == True