def test_table_imputer_2(): data = Table( os.path.join(os.path.dirname(__file__), 'data', 'Freedman.csv')) data1 = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman_imputer2.csv')) assert data.imputer('density', 800).equals(data1['density']) == True
def test_table_imputer_1(): data = Table( os.path.join(os.path.dirname(__file__), 'data', 'Freedman.csv')) data1 = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman_imputer1.csv')) assert data.imputer('population', 1000).equals(data1['population']) == True