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
Branches
Tags
2 merge requests!59Develop,!46Felix #56 advanced paddings
Pipeline #30917 passed
import keras
import tensorflow as tf
import numpy as np
import pytest
import mock
import os
from src.model_modules.advanced_paddings import *
......@@ -119,6 +115,7 @@ class TestReflectionPadding2D:
class TestSymmerticPadding2D:
@pytest.fixture
def input_x(self):
return keras.Input(shape=(10, 10, 3))
......@@ -158,7 +155,7 @@ class TestSymmerticPadding2D:
sym_pad = SymmetricPadding2D(padding=pad, name=layer_name)(input_x)
assert sym_pad.get_shape().as_list() == [None, 12, 10, 3]
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