class_docstr: Test functionality of SkorchDoctor using a simple model
class_docstr_tok: ["Test","functionality","of","SkorchDoctor","using","a","simple","model"]
code: def test_get_layer_names(self, doctor):
      layer_names = doctor.get_layer_names()
      expected = {
          "criterion": [],
          "module": ["lin0", "lin1", "softmax"],
      }
      assert layer_names == expected
  
imports: import itertools
  from unittest import mock
  import numpy
  import pytest
  import torch
  from torch import nn
label_desc: Use image node_id 18 for calling the TestSkorchDoctorSimple obj's underlying member method code with example usage: obj.test_get_layer_names(doctor) without return types