code: def test_infer_ten_classes():
      analyser = output_analysers.ClassificationAnalyser(name="a")
      dataset = test_utils.generate_one_hot_labels(
          dtype="dataset", num_classes=10
      )
  
      for data in dataset:
          analyser.update(data)
      analyser.finalize()
  
      assert analyser.num_classes == 10
  
file_path: autokeras/autokeras/analysers/output_analysers_test.py
imports: import numpy
  import pytest
  import tensorflow
  from autokeras import test_utils
  from autokeras.analysers import output_analysers
label_desc: Use image node_id 5 for calling a global function with example usage: test_infer_ten_classes() without return types