ts.torch_handler.unit_tests 封包

子封包

• ts.torch_handler.unit_tests.models 封包
  • 子模組
  • ts.torch_handler.unit_tests.models.base_model 模組
  • 模組內容
• ts.torch_handler.unit_tests.test_utils 封包
  • 子模組
  • ts.torch_handler.unit_tests.test_utils.mock_context 模組
  • 模組內容

子模組

ts.torch_handler.unit_tests.test_base_handler 模組

BaseHandler 類別的基本單元測試。確保它可以載入和執行範例模型

ts.torch_handler.unit_tests.test_base_handler.handler(base_model_context)

ts.torch_handler.unit_tests.test_base_handler.test_batch_handle(handler, base_model_context)

ts.torch_handler.unit_tests.test_base_handler.test_inference_with_profiler_works_with_custom_initialize_method(handler, base_model_context)

ts.torch_handler.unit_tests.test_base_handler.test_single_handle(handler, base_model_context)

ts.torch_handler.unit_tests.test_envelopes 模組

BaseHandler 類別的基本單元測試。確保它可以載入和執行範例模型

ts.torch_handler.unit_tests.test_envelopes.handle_fn(base_model_context)

ts.torch_handler.unit_tests.test_envelopes.test_binary(base_model_context)

ts.torch_handler.unit_tests.test_envelopes.test_body(handle_fn, base_model_context)

ts.torch_handler.unit_tests.test_envelopes.test_json(handle_fn, base_model_context)

ts.torch_handler.unit_tests.test_envelopes.test_json_batch(handle_fn, base_model_context)

ts.torch_handler.unit_tests.test_envelopes.test_json_double_batch(handle_fn, base_model_context)

更複雜的測試案例。 確保模型可以多工處理多個批次並傳回解多工的結果

ts.torch_handler.unit_tests.test_image_classifier 模組

ImageClassifier 類別的基本單元測試。 確保它可以載入和執行範例模型

ts.torch_handler.unit_tests.test_image_classifier.context(model_dir, model_name)

ts.torch_handler.unit_tests.test_image_classifier.handler(context)

ts.torch_handler.unit_tests.test_image_classifier.image_bytes()

ts.torch_handler.unit_tests.test_image_classifier.model_dir(tmp_path_factory, model_name)

ts.torch_handler.unit_tests.test_image_classifier.model_name()

ts.torch_handler.unit_tests.test_image_classifier.test_handle(context, image_bytes, handler)

ts.torch_handler.unit_tests.test_image_classifier.test_handle_explain(context, image_bytes, handler)

ts.torch_handler.unit_tests.test_image_segmenter 模組

ImageSegmenter 類別的基本單元測試。 確保它可以載入和執行範例模型

ts.torch_handler.unit_tests.test_image_segmenter.context(model_dir, model_name)

ts.torch_handler.unit_tests.test_image_segmenter.handler(context)

ts.torch_handler.unit_tests.test_image_segmenter.image_bytes()

ts.torch_handler.unit_tests.test_image_segmenter.model_dir(tmp_path_factory, model_name)

ts.torch_handler.unit_tests.test_image_segmenter.model_name()

ts.torch_handler.unit_tests.test_image_segmenter.test_handle(handler, context, image_bytes)

ts.torch_handler.unit_tests.test_mnist_kf 模組

ts.torch_handler.unit_tests.test_object_detector 模組

ObjectDetector 類別的基本單元測試。 確保它可以載入和執行範例模型

ts.torch_handler.unit_tests.test_object_detector.context(model_dir, model_name)

ts.torch_handler.unit_tests.test_object_detector.handler(context)

ts.torch_handler.unit_tests.test_object_detector.image_bytes()

ts.torch_handler.unit_tests.test_object_detector.model_dir(tmp_path_factory, model_name)

ts.torch_handler.unit_tests.test_object_detector.model_name()

ts.torch_handler.unit_tests.test_object_detector.test_handle(handler, context, image_bytes)

模組內容