Test for correct setting of hexagonal attribute when instantiating fr…

…om int number.
materialsproject · Jun 5, 2024 · 561dfef · 561dfef
1 parent ebe1246
commit 561dfef
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/symmetry/test_groups.py b/tests/symmetry/test_groups.py
@@ -205,6 +205,13 @@ def test_subgroup_supergroup(self):
         assert SpaceGroup("Pma2").is_subgroup(SpaceGroup("Pccm"))
         assert not SpaceGroup.from_int_number(229).is_subgroup(SpaceGroup.from_int_number(230))
 
+    def test_hexagonal(self):
+        for num in (146, 148, 155, 160, 161, 166, 167):
+            sg = SpaceGroup.from_int_number(num, hexagonal=False)
+            assert sg.hexagonal is False
+            sg = SpaceGroup.from_int_number(num, hexagonal=True)
+            assert sg.hexagonal is True
+
     def test_string(self):
         sg = SpaceGroup("R-3c")
         assert sg.to_latex_string() == r"R$\overline{3}$c"