class_docstr: Unit tests for the batch_transform class
class_docstr_tok: ["Unit","tests","for","the","batch_transform","class"]
code: def phaseshift_expand(tape):
      return tape.expand(stop_at=lambda obj: obj.name != "PhaseShift")
  
imports: import functools
  import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 3 for calling the TestBatchTransform obj's underlying member method code with example usage: obj.phaseshift_expand(tape) and returns: tape