From acd6c567c3e44b3cdee15278728f0d1f8e8918ea Mon Sep 17 00:00:00 2001
From: Jonathan Shimwell <drshimwell@gmail.com>
Date: Tue, 21 Jun 2022 14:19:45 +0000
Subject: [PATCH] added test for remove eleemnt

---
 tests/unit_tests/test_material.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/tests/unit_tests/test_material.py b/tests/unit_tests/test_material.py
index 02e53a28db..3a976da996 100644
--- a/tests/unit_tests/test_material.py
+++ b/tests/unit_tests/test_material.py
@@ -38,6 +38,17 @@ def test_remove_nuclide():
     assert m.nuclides[1].percent == 2.0
 
 
+def test_remove_elements():
+    """Test removing elements."""
+    m = openmc.Material()
+    for elem, percent in [('Li', 1.0), ('Be', 1.0)]:
+        m.add_element(elem, percent)
+    m.remove_nuclide('Li')
+    assert len(m.nuclides) == 1
+    assert m.nuclides == ['Be9']
+    assert m.nuclides[0].percent == 1.0
+
+
 def test_elements():
     """Test adding elements."""
     m = openmc.Material()