1// Copyright 2018 Google Inc. All rights reserved. 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at: 6// 7// http://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14 15#import "tensorflow/lite/objc/apis/TFLQuantizationParameters.h" 16 17#import <XCTest/XCTest.h> 18 19#import "tensorflow/lite/objc/sources/TFLQuantizationParameters+Internal.h" 20 21NS_ASSUME_NONNULL_BEGIN 22 23/** Test scale of quantization parameters. */ 24static const float kTestScale = 2.0; 25 26/** Test zero point of quantization parameters. */ 27static const int32_t kTestZeroPoint = 128; 28 29/** 30 * Unit tests for TFLQuantizationParameters. 31 */ 32@interface TFLQuantizationParametersTests : XCTestCase 33@end 34 35@implementation TFLQuantizationParametersTests 36 37#pragma mark - Tests 38 39- (void)testInitWithScaleAndZeroPoint { 40 TFLQuantizationParameters *params = 41 [[TFLQuantizationParameters alloc] initWithScale:kTestScale zeroPoint:kTestZeroPoint]; 42 XCTAssertEqual(params.scale, kTestScale); 43 XCTAssertEqual(params.zeroPoint, kTestZeroPoint); 44} 45 46@end 47 48NS_ASSUME_NONNULL_END 49