From 6001673ead6337cc1cb75b8ab589e5808ae71c86 Mon Sep 17 00:00:00 2001 From: s184364 <s184364@student.dtu.dk> Date: Tue, 12 Dec 2023 09:06:46 +0100 Subject: [PATCH] Solved: 2 tests for gradio app launch failed on windows + investigation of pytest warnings --- qim3d/tests/gui/test_annotation_tool.py | 11 ++++++----- qim3d/tests/gui/test_iso3d.py | 11 ++++++----- qim3d/utils/internal_tools.py | 1 - qim3d/viz/img.py | 6 +++--- 4 files changed, 15 insertions(+), 14 deletions(-) diff --git a/qim3d/tests/gui/test_annotation_tool.py b/qim3d/tests/gui/test_annotation_tool.py index 65c02931..2b80b9fb 100644 --- a/qim3d/tests/gui/test_annotation_tool.py +++ b/qim3d/tests/gui/test_annotation_tool.py @@ -9,14 +9,15 @@ def test_starting_class(): assert app.title == "Annotation tool" +def start_server(ip, port): + app = qim3d.gui.annotation_tool.Interface() + app.launch(server_name=ip, server_port=port) + + def test_app_launch(): - ip = "0.0.0.0" + ip = "localhost" port = 65432 - def start_server(ip, port): - app = qim3d.gui.annotation_tool.Interface() - app.launch(server_name=ip, server_port=port) - proc = multiprocessing.Process(target=start_server, args=(ip, port)) proc.start() diff --git a/qim3d/tests/gui/test_iso3d.py b/qim3d/tests/gui/test_iso3d.py index 4e10123d..864ede39 100644 --- a/qim3d/tests/gui/test_iso3d.py +++ b/qim3d/tests/gui/test_iso3d.py @@ -9,14 +9,15 @@ def test_starting_class(): assert app.title == "Isosurfaces for 3D visualization" +def start_server(ip, port): + app = qim3d.gui.iso3d.Interface() + app.launch(server_name=ip, server_port=port) + + def test_app_launch(): - ip = "0.0.0.0" + ip = "localhost" port = 65432 - def start_server(ip, port): - app = qim3d.gui.iso3d.Interface() - app.launch(server_name=ip, server_port=port) - proc = multiprocessing.Process(target=start_server, args=(ip, port)) proc.start() diff --git a/qim3d/utils/internal_tools.py b/qim3d/utils/internal_tools.py index c8340d45..a4f6ede0 100644 --- a/qim3d/utils/internal_tools.py +++ b/qim3d/utils/internal_tools.py @@ -6,7 +6,6 @@ import outputformat as ouf import matplotlib.pyplot as plt import matplotlib import numpy as np -import socket import os import shutil import requests diff --git a/qim3d/viz/img.py b/qim3d/viz/img.py index 83edde50..e3648f54 100644 --- a/qim3d/viz/img.py +++ b/qim3d/viz/img.py @@ -1,7 +1,7 @@ """ Provides a collection of visualization functions.""" import matplotlib.pyplot as plt from matplotlib.colors import LinearSegmentedColormap -from matplotlib import cm +from matplotlib import colormaps import torch import numpy as np from qim3d.io.logger import log @@ -56,7 +56,7 @@ def grid_overview(data, num_images=7, cmap_im="gray", cmap_segm="viridis", alpha num_images = len(data) # Adapt segmentation cmap so that background is transparent - colors_segm = cm.get_cmap(cmap_segm)(np.linspace(0, 1, 256)) + colors_segm = colormaps.get_cmap(cmap_segm)(np.linspace(0, 1, 256)) colors_segm[:128, 3] = 0 custom_cmap = LinearSegmentedColormap.from_list("CustomCmap", colors_segm) @@ -142,7 +142,7 @@ def grid_pred(in_targ_preds, num_images=7, cmap_im="gray", cmap_segm="viridis", inputs,targets,preds = [items[:num_images] for items in in_targ_preds] # Adapt segmentation cmap so that background is transparent - colors_segm = cm.get_cmap(cmap_segm)(np.linspace(0, 1, 256)) + colors_segm = colormaps.get_cmap(cmap_segm)(np.linspace(0, 1, 256)) colors_segm[:128, 3] = 0 custom_cmap = LinearSegmentedColormap.from_list("CustomCmap", colors_segm) -- GitLab