Exemplos de MLModelSerializer.save em Python

Exemplo n.º 1

Arquivo: test_mlmodel_serializer.py Projeto: rennym19/ml-playgrounds

    def test_save(self, mock_is_instance, mock_save):
        mock_is_instance.return_value = True
        mock_save.return_value = MockModel(uid=ObjectId())

        serializer = MLModelSerializer()
        saved_instance = serializer.save(MockModel())

        self.assertIsNotNone(saved_instance.uid)

Exemplo n.º 2

Arquivo: test_mlmodel_serializer.py Projeto: rennym19/ml-playgrounds

    def test_save_invalid_instance(self):
        serializer = MLModelSerializer()

        with self.assertRaises(TypeError):
            serializer.save(MockModel())