def test_present_embeddings_run_server(self): def sweded_which(prog): return None which = shutil.which shutil.which = sweded_which browser = os.getenv("BROWSER", "") os.environ["BROWSER"] = "" try: with tempfile.TemporaryDirectory( prefix="sourced.ml-test-") as tmpdir: with captured_output() as (stdout, _, _): present_embeddings(tmpdir, True, ["one"], [str(i) for i in range(5)], [(i, i) for i in range(5)]) with open(os.path.join(tmpdir, "id2vec.json")) as fin: json.load(fin) with open(os.path.join(tmpdir, "id2vec_meta.tsv")) as fin: self.assertEqual(fin.read(), "0\n1\n2\n3\n4\n") with open(os.path.join(tmpdir, "id2vec_data.tsv")) as fin: self.assertEqual(fin.read(), "0\t0\n1\t1\n2\t2\n3\t3\n4\t4\n") self.assertIn( "\thttp://projector.tensorflow.org/?config=http://0.0.0.0:8000/id2vec.json\n", stdout.getvalue()) finally: shutil.which = which os.environ["BROWSER"] = browser web_server.stop()
def test_wait(self): web_server.start() try: attempts, result = self.wait_for_web_server() self.assertTrue(attempts < self.MAX_ATTEMPTS or result == 0) self.assertTrue(web_server.running) except: # noqa web_server.stop() raise os.environ["PROJECTOR_SERVER_TIME"] = "0" wait() self.assertFalse(web_server.running) web_server.start() try: attempts, result = self.wait_for_web_server() self.assertTrue(attempts < self.MAX_ATTEMPTS or result == 0) self.assertTrue(web_server.running) finally: web_server.stop()
def test_stop(self): web_server.stop() # dummy test to avoid partially covered line in CI self.assertFalse(web_server.running) web_server.start() web_server.stop() self.assertFalse(web_server.running)