@pytest.mark.skipif((not _IMAGE_TESTING), reason="image libraries aren't installed.")
def test_predict_numpy():
    img = np.ones((1, 3, 64, 64))
    model = SemanticSegmentation(2, backbone='mobilenetv3_large_100')
    data_pipe = DataPipeline(preprocess=SemanticSegmentationPreprocess(num_classes=1))
    out = model.predict(img, data_source='numpy', data_pipeline=data_pipe)
    assert isinstance(out[0], list)
    assert (len(out[0]) == 64)
    assert (len(out[0][0]) == 64)
@pytest.mark.skipif((not _IMAGE_TESTING), reason="image libraries aren't installed.")
def test_predict_numpy():
    img = np.ones((1, 3, 64, 64))
    model = SemanticSegmentation(2, backbone='mobilenetv3_large_100')
    data_pipe = DataPipeline(preprocess=SemanticSegmentationPreprocess(num_classes=1))
    out = model.predict(img, data_source='numpy', data_pipeline=data_pipe)
    assert isinstance(out[0], list)
    assert (len(out[0]) == 64)
    assert (len(out[0][0]) == 64)