Skip to content
Snippets Groups Projects
Commit f2ccaa3d authored by Felix Kleinert's avatar Felix Kleinert
Browse files

#56 first test implementations

parent 90d6aa25
Branches
Tags
2 merge requests!59Develop,!46Felix #56 advanced paddings
Pipeline #30905 passed
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))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment