def test_check_columns_correct():
'''
Tests check_columns correctly checks items summing to 1.0 returns true
'''
df = pd.DataFrame([0.1, 0.9])
assert check_columns(df)
def test_check_columns_correct():
'''
Tests check_columns correctly checks items summing to 1.0 returns true
'''
df = pandas.DataFrame(data =[[0.2,0.5],[0.8,0.5]])
assert check_columns(df) is True
def test_check_columns_correct():
'''FIXME: Implement this test
Tests check_columns correctly checks items summing to 1.0 returns true
'''
df = pandas.DataFrame([0.5, 0.5])
assert check_columns(df) is True
def test_check_columns_incorrect():
df = pandas.DataFrame([0.1, 0.8])
assert check_columns(df) is False
def test_check_columns_correct():
df = pandas.DataFrame([0.2, 0.8])
assert check_columns(df) is True
