def test_get_client_class(self):
server = JubaRPCServer(self.node, CLASSIFIER, [])
self.assertEqual(server.get_client_class(), jubatus.classifier.client.Classifier)
server = JubaRPCServer(self.node, NEAREST_NEIGHBOR, [])
self.assertEqual(server.get_client_class(), jubatus.nearest_neighbor.client.NearestNeighbor)
def setUp(self):
self.node = JubaNode('127.0.0.1', [12345], None, '/tmp', [])
self.instance = JubaRPCServer(self.node, CLASSIFIER, [])
self.stub_instance = JubaRPCServerStub(self.node)
