diff --git a/test/test_model_modules/test_advanced_paddings.py b/test/test_model_modules/test_advanced_paddings.py index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..03782c3fa6a5de7d571fcf1ec76d6448a044f552 100644 --- a/test/test_model_modules/test_advanced_paddings.py +++ b/test/test_model_modules/test_advanced_paddings.py @@ -0,0 +1,46 @@ +import keras +import numpy as np +import pytest +import mock +import os + +from src.model_modules.advanced_paddings import * + + +class TestPadUtils: + + def test_get_padding_for_same(self): + with pytest.raises(ValueError): + pad_utils.get_padding_for_same((-1, 2)) + pad_utils.get_padding_for_same((1, 0)) + + kernel = (1, 1) + pad = pad_utils.get_padding_for_same(kernel) + assert pad == (0, 0) + assert isinstance(pad, tuple) + assert isinstance(pad[0], int) and isinstance(pad[1], int) + assert not isinstance(pad[0], np.int64) + with pytest.raises(NotImplementedError): + pad_utils.get_padding_for_same(kernel, strides=2) + + kernel = (3, 1) + pad = pad_utils.get_padding_for_same(kernel) + assert pad == (1, 0) + assert isinstance(pad, tuple) + assert isinstance(pad[0], int) and isinstance(pad[1], int) + assert not isinstance(pad[0], np.int64) + + kernel = (2, 2) + with pytest.raises(NotImplementedError): + pad_utils.get_padding_for_same(kernel) + + kernel = (3, 3, 3) + pad = pad_utils.get_padding_for_same(kernel) + assert pad == (1, 1, 1) + assert isinstance(pad, tuple) + assert isinstance(pad[0], int) and isinstance(pad[1], int) + assert not (isinstance(pad[0], np.int64) and isinstance(pad[0], np.int64)) + + + +