def test_remove_records(): df = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman.csv')) df1 = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman_remove_records.csv')) check = Check() assert check.remove_records(df).equals(df1) == True
def test_remove_records_1(): df = pd.read_csv( os.path.join(os.path.dirname(__file__), 'data', 'Freedman.csv')) check = Check() assert len(check.remove_records(df)) == 110
def test_remove_records_2(): df = pd.read_csv(os.path.join(os.path.dirname(__file__), 'data', 'msleep_ggplot.csv')) check = Check() assert len(check.remove_records(df)) == 61