Add test for Kpoint equality

materialsproject · Feb 23, 2024 · 50efc02 · 50efc02
1 parent 7c47053
commit 50efc02
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/electronic_structure/test_bandstructure.py b/tests/electronic_structure/test_bandstructure.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import copy
 import json
 import unittest
 
@@ -27,6 +28,13 @@ def setUp(self):
         self.lattice = Lattice.cubic(10.0)
         self.kpoint = Kpoint([0.1, 0.4, -0.5], self.lattice, label="X")
 
+    def test_eq(self):
+        assert self.kpoint == self.kpoint
+        assert self.kpoint == copy.deepcopy(self.kpoint)
+        assert self.kpoint != Kpoint([0.1, 0.4, -0.5], self.lattice, label="Y")
+        assert self.kpoint != Kpoint([0.1, 0.4, -0.6], self.lattice, label="X")
+        assert self.kpoint != Kpoint([0.1, 0.4, -0.5], Lattice.cubic(20.0), label="X")
+
     def test_properties(self):
         assert list(self.kpoint.frac_coords) == [0.1, 0.4, -0.5]
         assert self.kpoint.a == 0.1