def test_values_only_without_values(): with pytest.raises(main.EmptyQueryResult): main.values_only(sum)([]) with pytest.raises(main.EmptyQueryResult): main.values_only(sum)([None, None, None])
def test_values_only_with_values(): assert main.values_only(sum)([1, 2, 3]) == 6 assert main.values_only(sum)([1, None, 3]) == 4