From 31d2b876cf0b101b31d61c08ab2c3605086bf96e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Radim=20=C5=98eh=C5=AF=C5=99ek?= Date: Mon, 27 Jul 2020 11:56:57 +0200 Subject: [PATCH] increase test timeout for Visdom server --- gensim/test/test_lda_callback.py | 17 +++++------------ 1 file changed, 5 insertions(+), 12 deletions(-) diff --git a/gensim/test/test_lda_callback.py b/gensim/test/test_lda_callback.py index 621dd8f369..e5234dfe38 100644 --- a/gensim/test/test_lda_callback.py +++ b/gensim/test/test_lda_callback.py @@ -36,23 +36,16 @@ def setUp(self): self.port = 8097 def testCallbackUpdateGraph(self): - - # Popen have no context-manager in 2.7, for this reason - try/finally. - try: - # spawn visdom.server - proc = subprocess.Popen(['python', '-m', 'visdom.server', '-port', str(self.port)]) - + with subprocess.Popen(['python', '-m', 'visdom.server', '-port', str(self.port)]) as proc: # wait for visdom server startup (any better way?) - time.sleep(3) - viz = Visdom(server=self.host, port=self.port) + for attempt in range(5): + time.sleep(1.0) # seconds + if viz.check_connection(): + break assert viz.check_connection() - - # clear screen viz.close() - self.model.update(self.corpus) - finally: proc.kill()