fix: refine test code to avoid if logic

Author: stevenhua0320

tests/test_structure.py

@@ -122,7 +122,6 @@ def test___copy__(self):
 
     def test_add_new_atom(self):
         s_lat = Lattice()
-        other_lat = Lattice()
         expected = Structure(lattice=s_lat)
 
         length = len(expected)
@@ -131,26 +130,20 @@ def test_add_new_atom(self):
         assert len(expected) == actual
 
         object = expected[-1]
-        if hasattr(object, "element"):
-            assert object.element == "C"
-        if hasattr(object, "xyz"):
-            assert numpy.allclose(object.xyz, [0.1, 0.2, 0.3])
+        assert object.element == "C"
+        assert numpy.allclose(object.xyz, [0.1, 0.2, 0.3])
 
     def test_addNewAtom(self):
         s_lat = Lattice()
-        other_lat = Lattice()
         expected = Structure(lattice=s_lat)
 
         length = len(expected)
         expected.addNewAtom(atype="C", xyz=[0.1, 0.2, 0.3])
         actual = length + 1
         assert len(expected) == actual
-
         object = expected[-1]
-        if hasattr(object, "element"):
-            assert object.element == "C"
-        if hasattr(object, "xyz"):
-            assert numpy.allclose(object.xyz, [0.1, 0.2, 0.3])
+        assert object.element == "C"
+        assert numpy.allclose(object.xyz, [0.1, 0.2, 0.3])
 
     def test_assignUniqueLabels(self):
         """Check Structure.assignUniqueLabels()"""