fl_server_core.tests.test_model ¶

class ModelTest(TestCase):

    def test_is_global_model(self):
        model = Dummy.create_model()
        self.assertTrue(model.is_global_model())
        self.assertFalse(model.is_local_model())

    def test_is_local_model(self):
        model = Dummy.create_model_update()
        self.assertFalse(model.is_global_model())
        self.assertTrue(model.is_local_model())

    def test_clone_model(self):
        model1 = Dummy.create_model()
        model2 = clone_model(model1)
        model2.description = "This Model was duplicated"
        model1.description = "This is the original"

        self.assertEqual(model1.name, model2.name)
        self.assertEqual(model2.description, "This Model was duplicated")
        self.assertEqual(model1.description, "This is the original")
        self.assertEqual(model1.owner, model2.owner)
        self.assertEqual(model1.round, model2.round)
        self.assertEqual(model1.weights, model2.weights)
        self.assertEqual(model1.preprocessing, model2.preprocessing)