From f2ccaa3d77c9552094845c7b9bcc65173a7ac0fe Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Mon, 2 Mar 2020 12:01:59 +0100
Subject: [PATCH] #56 first test implementations

---
 .../test_advanced_paddings.py                 | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/test/test_model_modules/test_advanced_paddings.py b/test/test_model_modules/test_advanced_paddings.py
index e69de29b..03782c3f 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))
+
+
+
+
-- 
GitLab