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

update some tests for advanced paddings (#56)

parent f38b1cfd
No related branches found
No related tags found
2 merge requests!59Develop,!46Felix #56 advanced paddings
Pipeline #30917 passed
import keras import keras
import tensorflow as tf
import numpy as np
import pytest import pytest
import mock
import os
from src.model_modules.advanced_paddings import * from src.model_modules.advanced_paddings import *
...@@ -119,6 +115,7 @@ class TestReflectionPadding2D: ...@@ -119,6 +115,7 @@ class TestReflectionPadding2D:
class TestSymmerticPadding2D: class TestSymmerticPadding2D:
@pytest.fixture @pytest.fixture
def input_x(self): def input_x(self):
return keras.Input(shape=(10, 10, 3)) return keras.Input(shape=(10, 10, 3))
...@@ -158,7 +155,7 @@ class TestSymmerticPadding2D: ...@@ -158,7 +155,7 @@ class TestSymmerticPadding2D:
sym_pad = SymmetricPadding2D(padding=pad, name=layer_name)(input_x) sym_pad = SymmetricPadding2D(padding=pad, name=layer_name)(input_x)
assert sym_pad.get_shape().as_list() == [None, 12, 10, 3] assert sym_pad.get_shape().as_list() == [None, 12, 10, 3]
assert sym_pad.name == 'SymPad_3x1/MirrorPad:0' assert sym_pad.name == 'SymPad_3x1/MirrorPad:0'
print(123)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment