def test_find_pooling_constant_bad_divisor():
"""Test error when trying to downsample to a non-divisor of the features"""
with pytest.raises(ValueError):
_find_pooling_constant(POOLING_FEATURES, 40)
with pytest.raises(ValueError):
_find_pooling_constant(POOLING_FEATURES, 0)
def test_find_pooling_constant():
"""Test that pooling constant given correct answer with good inputs"""
assert _find_pooling_constant(POOLING_FEATURES, 6) == 10
def test_find_pooling_constant_upsample():
"""Test error when trying to upsample"""
with pytest.raises(ValueError):
_find_pooling_constant(POOLING_FEATURES, 120)