class_docstr: Do a sanity check on the models.
class_docstr_tok: ["Do","a","sanity","check","on","the","models","."]
docstr: Test the models on only 1 epoch as a sanity check before merge.
docstr_tok: ["Test","the","models","on","only","1","epoch","as","a","sanity","check","before","merge","."]
code: def test_model(self, model_name, nncf, category="shapes", path=""):
      """Test the models on only 1 epoch as a sanity check before merge."""
      with tempfile.TemporaryDirectory() as project_path:
          # Train test
          config, datamodule, model, trainer = setup_model_train(
              model_name,
              dataset_path=path,
              project_path=project_path,
              nncf=nncf,
              category=category,
              fast_run=True,
          )
          results = trainer.test(model=model, datamodule=datamodule)[0]
  
          # Test model load
          model_load_test(config, datamodule, results)
  
imports: import tempfile
  import pytest
  from tests.helpers.dataset import TestDataset
  from tests.helpers.model import model_load_test, setup_model_train
label_desc: Use image node_id 1 for calling the TestModel obj's underlying member method code with example usage: obj.test_model(model_name, nncf, category, path) without return types