code: def test_get_atom_mass():
      atomic_number = 41
      atomic_mass = get_atom_mass(atomic_number)
      assert np.allclose(atomic_mass, 92.9064 * 1822.888486209)
  
imports: import numpy
  import pytest
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_atom_mass() without return types