doc-ai / tensorio-ios

Tensor/IO for iOS, with support for on-device inference and training with TensorFlow and TensorFlow Lite
Other
31 stars 10 forks source link

Clean up expectations and assertions #109

Closed phildow closed 5 years ago

phildow commented 5 years ago
MockSessionDataTask *task = (MockSessionDataTask*)[client GETTaskWithTaskId:@"task-id" callback:^(TIOFleaTask * _Nullable task, NSError * _Nullable error) {
  [expectation fulfill];

  XCTAssertNil(error);
  XCTAssertNotNil(task);

  XCTAssertEqualObjects(task.modelId, @"model-id");
  XCTAssertEqualObjects(task.hyperparametersId, @"hyperparameters-id");
  XCTAssertEqualObjects(task.checkpointId, @"checkpoint-id");
  // ...
}];