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)