code: def test_editable_module_pure_function():
      from deepchem.utils.differentiation_utils import EditableModule
      from deepchem.utils.differentiation_utils.pure_function import (
          EditableModulePureFunction,
      )
  
      class A(EditableModule):
          def __init__(self, a):
              self.b = a * a
  
          def mult(self, x):
              return self.b * x
  
          def getparamnames(self, methodname, prefix=""):
              if methodname == "mult":
                  return [prefix + "b"]
              else:
                  raise KeyError()
  
      B = A(4)
      m = EditableModulePureFunction(B, B.mult)
      m.set_objparams([3])
      assert m(2) == 6
  
imports: import pytest
label_desc: Use image node_id 32 for calling a global function with example usage: test_editable_module_pure_function() and returns: unknown, list