code: def test_get_layer_set(layer_set: str, layers: List[str]) -> None:
      """Tests assignment of individual layers to the layer set.
  
      Args:
          layer_set: layer set (e.g. linear, rnn_base)
          layers: non-exhaustive list of layers that belong to the layer_set
      """
      assert all(get_layer_set(layer) == layer_set for layer in layers)
  
file_path: opacus/benchmarks/tests/test_utils.py
imports: import os
  import pickle
  import shutil
  from typing import Any, Dict, List, Tuple
  import pytest
  import torch
  from helpers import get_n_byte_tensor, skipifnocuda
  from benchmarks.utils import get_layer_set, get_path, reset_peak_memory_stats, save_results
label_desc: Use image node_id 1 for calling a global function with example usage: test_get_layer_set(layer_set, layers) without return types