From 3603b3fdc183eeedff1ed2de3b00b4ad132e3ea9 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Mon, 18 May 2020 20:31:31 +0200
Subject: [PATCH] [test.test_summary] Fix imports
---
bob/ip/binseg/test/test_summary.py | 20 ++++++++++----------
1 file changed, 10 insertions(+), 10 deletions(-)
diff --git a/bob/ip/binseg/test/test_summary.py b/bob/ip/binseg/test/test_summary.py
index b29bfcc8..512bb93f 100644
--- a/bob/ip/binseg/test/test_summary.py
+++ b/bob/ip/binseg/test/test_summary.py
@@ -4,11 +4,11 @@
import os
import unittest
-from ..modeling.driu import build_driu
-from ..modeling.driuod import build_driuod
-from ..modeling.hed import build_hed
-from ..modeling.unet import build_unet
-from ..modeling.resunet import build_res50unet
+from ..models.driu import driu
+from ..models.driu_od import driu_od
+from ..models.hed import hed
+from ..models.unet import unet
+from ..models.resunet import resunet50
from ..utils.summary import summary
@@ -18,31 +18,31 @@ class Tester(unittest.TestCase):
"""
def test_summary_driu(self):
- model = build_driu()
+ model = driu()
s, param = summary(model)
self.assertIsInstance(s, str)
self.assertIsInstance(param, int)
def test_summary_driuod(self):
- model = build_driuod()
+ model = driu_od()
s, param = summary(model)
self.assertIsInstance(s, str)
self.assertIsInstance(param, int)
def test_summary_hed(self):
- model = build_hed()
+ model = hed()
s, param = summary(model)
self.assertIsInstance(s, str)
self.assertIsInstance(param, int)
def test_summary_unet(self):
- model = build_unet()
+ model = unet()
s, param = summary(model)
self.assertIsInstance(s, str)
self.assertIsInstance(param, int)
def test_summary_resunet(self):
- model = build_res50unet()
+ model = resunet()
s, param = summary(model)
self.assertIsInstance(s, str)
self.assertIsInstance(param, int)
--
GitLab