class_docstr: Test functionality of SkorchDoctor using a simple model
class_docstr_tok: ["Test","functionality","of","SkorchDoctor","using","a","simple","model"]
code: def doctor(self, module_cls, net_cls, doctor_cls, data, custom_split):
      net = net_cls(
          module_cls,
          max_epochs=3,
          batch_size=32,
          train_split=custom_split,
      )
      doctor = doctor_cls(net)
      doctor.fit(*data)
      return doctor
  
imports: import itertools
  from unittest import mock
  import numpy
  import pytest
  import torch
  from torch import nn
label_desc: Use image node_id 6 for calling the TestSkorchDoctorSimple obj's underlying member method code with example usage: obj.doctor(module_cls, net_cls, doctor_cls, data, custom_split) and returns: doctor