def test_2col_dimensions(): ''' return False when wrong number of columns present ''' with patch.object(RunSnpDetection, "__init__", lambda x: None): detect_obj = RunSnpDetection() tab = pandas.DataFrame({'A': [1], 'B': [2]}) assert detect_obj.three_cols(tab) == False
def test_3col_dimensions(): ''' return True when correct number of columns ''' with patch.object(RunSnpDetection, "__init__", lambda x: None): detect_obj = RunSnpDetection() tab = pandas.DataFrame({'A': [1], 'B': [2], 'C': [3]}) assert detect_obj.three_cols(tab)