From 6ee128172f90501d378bd361259fdcd1c4c98018 Mon Sep 17 00:00:00 2001
From: "v.gramlich1" <v.gramlichfz-juelich.de>
Date: Mon, 28 Jun 2021 11:40:46 +0200
Subject: [PATCH] Updated test to test on IntelliO3_ts_architecture instead of
the old MyPaperModel
---
test/test_model_modules/test_model_class.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/test/test_model_modules/test_model_class.py b/test/test_model_modules/test_model_class.py
index cbff4cec..b05fd990 100644
--- a/test/test_model_modules/test_model_class.py
+++ b/test/test_model_modules/test_model_class.py
@@ -1,14 +1,14 @@
import keras
import pytest
-from mlair.model_modules.model_class import MyPaperModel
+from mlair.model_modules.model_class import IntelliO3_ts_architecture
-class TestMyPaperModel:
+class TestIntelliO3_ts_architecture:
@pytest.fixture
def mpm(self):
- return MyPaperModel(input_shape=[(7, 1, 9)], output_shape=[(4,)])
+ return IntelliO3_ts_architecture(input_shape=[(7, 1, 9)], output_shape=[(4,)])
def test_init(self, mpm):
# check if loss number of loss functions fit to model outputs
--
GitLab