def setUpClass(self): glconnect.stop() auth_token = 'graphlab_awesome' self.server = start_test_tcp_server(auth_token=auth_token) glconnect.launch(self.server.get_server_addr(), auth_token=auth_token) self.tempfile = tempfile.NamedTemporaryFile().name (self.graph, self.sframe, self.model) = create_test_objects()
def test_launch_to_tcp(self): auth_token = 'graphlab_awesome' tcp_server = start_test_tcp_server(auth_token) #launch with remote server tcp address glconnect.launch(tcp_server.get_server_addr(), auth_token=auth_token) self.assertTrue(glconnect.is_connected()) self.assertTrue(isinstance(glconnect.get_server(), server.RemoteServer)) glconnect.stop() self.assertFalse(glconnect.is_connected()) tcp_server.stop()