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