code: def test_impute_hetatm():
      new = ppdb.impute_element(records=["HETATM"])
      assert new["HETATM"]["element_symbol"][1] == "N"
      assert new["HETATM"]["element_symbol"][10] == "O"
      assert new["ATOM"]["element_symbol"][1] == ""
      assert new["ATOM"]["element_symbol"][10] == ""
  
file_path: biopandas/biopandas/pdb/tests/test_impute.py
imports: from biopandas.pdb import PandasPdb
  import os
  import numpy
  import pandas
  from nose.tools import raises
label_desc: Use image node_id 1 for calling a global function with example usage: test_impute_hetatm() without return types