36 lines
1.1 KiB
Swift
36 lines
1.1 KiB
Swift
//
|
|
// Copyright Amazon.com Inc. or its affiliates.
|
|
// All Rights Reserved.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
import XCTest
|
|
import Amplify
|
|
@testable import CoreMLPredictionsPlugin
|
|
|
|
class CoreMLPredictionsPluginTestBase: XCTestCase {
|
|
|
|
var coreMLPredictionsPlugin: CoreMLPredictionsPlugin!
|
|
|
|
var naturalLanguageBehavior: MockCoreMLNaturalLanguageAdapter!
|
|
|
|
var visionBehavior: MockCoreMLVisionAdapter!
|
|
|
|
var speechBehavior: MockCoreMLSpeechAdapter!
|
|
|
|
var queue: MockOperationQueue!
|
|
|
|
override func setUp() {
|
|
coreMLPredictionsPlugin = CoreMLPredictionsPlugin()
|
|
naturalLanguageBehavior = MockCoreMLNaturalLanguageAdapter()
|
|
visionBehavior = MockCoreMLVisionAdapter()
|
|
speechBehavior = MockCoreMLSpeechAdapter()
|
|
queue = MockOperationQueue()
|
|
coreMLPredictionsPlugin.configure(naturalLanguageBehavior: naturalLanguageBehavior,
|
|
visionBehavior: visionBehavior,
|
|
speechBehavior: speechBehavior,
|
|
queue: queue)
|
|
}
|
|
}
|