Source code for florist.tests.unit.api.clients.test_common

from florist.api.clients.common import Client
from florist.api.clients.mnist import MnistClient, MnistFedProxClient



[docs]
def test_class_for_client():
    assert Client.class_for_client(Client.MNIST) == MnistClient
    assert Client.class_for_client(Client.MNIST_FEDPROX) == MnistFedProxClient




[docs]
def test_list():
    assert Client.list() == [Client.MNIST.value, Client.MNIST_FEDPROX.value]