From 4e21e398c83f0997fc20b7c4a08952cfc73b482b Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Mon, 19 Feb 2018 10:19:02 -0600
Subject: [PATCH] Move density_to_mat to example_geometry.py

---
 openmc/deplete/openmc_wrapper.py | 21 ---------------------
 scripts/example_geometry.py      | 23 ++++++++++++++++++++++-
 2 files changed, 22 insertions(+), 22 deletions(-)

diff --git a/openmc/deplete/openmc_wrapper.py b/openmc/deplete/openmc_wrapper.py
index 7cdfbd4d8..c8a3bcca9 100644
--- a/openmc/deplete/openmc_wrapper.py
+++ b/openmc/deplete/openmc_wrapper.py
@@ -657,24 +657,3 @@ class OpenMCOperator(Operator):
         volume = {k: v for d in volume_list for k, v in d.items()}
 
         return volume, nuc_list, burn_list, self.burnable_mats
-
-
-def density_to_mat(dens_dict):
-    """Generates an OpenMC material from a cell ID and self.number_density.
-
-    Parameters
-    ----------
-    m_id : int
-        Cell ID.
-    Returns
-    -------
-    openmc.Material
-        The OpenMC material filled with nuclides.
-    """
-
-    mat = openmc.Material()
-    for key in dens_dict:
-        mat.add_nuclide(key, 1.0e-24*dens_dict[key])
-    mat.set_density('sum')
-
-    return mat
diff --git a/scripts/example_geometry.py b/scripts/example_geometry.py
index 09ce0576f..ca10c1f72 100644
--- a/scripts/example_geometry.py
+++ b/scripts/example_geometry.py
@@ -9,7 +9,28 @@ import math
 
 import numpy as np
 import openmc
-from openmc.deplete import density_to_mat
+
+
+def density_to_mat(dens_dict):
+    """Generates an OpenMC material from a cell ID and self.number_density.
+
+    Parameters
+    ----------
+    dens_dict : dict
+        Dictionary mapping nuclide names to densities
+
+    Returns
+    -------
+    openmc.Material
+        The OpenMC material filled with nuclides.
+
+    """
+    mat = openmc.Material()
+    for key in dens_dict:
+        mat.add_nuclide(key, 1.0e-24*dens_dict[key])
+    mat.set_density('sum')
+
+    return mat
 
 
 def generate_initial_number_density():