Skip to content
Snippets Groups Projects
Commit 3603b3fd authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[test.test_summary] Fix imports

parent 03838cae
No related branches found
No related tags found
1 merge request!13Cleanup model implementation
Pipeline #39944 failed
......@@ -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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment