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
