Exemple #1
0
def test_mxnet_transformer_initialize_with_model(transformer_initialize):
    t = MXNetTransformer(model=Mock())
    t.initialize()

    transformer_initialize.assert_not_called()
Exemple #2
0
def test_mxnet_transformer_initialize_without_model(transformer_initialize):
    t = MXNetTransformer()
    t.initialize()

    transformer_initialize.assert_called_once()