diff --git a/test/test_model_modules/test_advanced_paddings.py b/test/test_model_modules/test_advanced_paddings.py index 1a565bc5992eb272d7406675cceac10d509ade69..7fec96dae6e23e46c7b313f841fb468b8b2af98f 100644 --- a/test/test_model_modules/test_advanced_paddings.py +++ b/test/test_model_modules/test_advanced_paddings.py @@ -1,9 +1,5 @@ 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) +