From c56b3850cd94e1c874c931697bfa3cb20b1cd002 Mon Sep 17 00:00:00 2001
From: Christian Hill <xn.hill@gmail.com>
Date: Wed, 2 Feb 2022 17:43:32 +0100
Subject: [PATCH] Add Gamma states to MolecularTermSymbol class

---
 src/pyvalem/states/molecular_term_symbol.py | 12 +++++++++++-
 tests/test_molecular_term_symbol.py         | 12 ++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/pyvalem/states/molecular_term_symbol.py b/src/pyvalem/states/molecular_term_symbol.py
index 4ff6983..aa50cac 100644
--- a/src/pyvalem/states/molecular_term_symbol.py
+++ b/src/pyvalem/states/molecular_term_symbol.py
@@ -39,6 +39,12 @@
     "PHI",
     "PHIg",
     "PHIu",
+    "Γ",
+    "Γg",
+    "Γu",
+    "GAMMA",
+    "GAMMAg",
+    "GAMMAu",
     "A",
     "A'",
     'A"',
@@ -140,9 +146,13 @@
     "PHI": "Φ",
     "PHIg": "Φg",
     "PHIu": "Φu",
+    "GAMMA": "Γ",
+    "GAMMAg": "Γg",
+    "GAMMAu": "Γu",
 }
 
-latex_terms = {"Σ": r"\Sigma", "Π": r"\Pi", "Δ": r"\Delta", "Φ": r"\Phi"}
+latex_terms = {"Σ": r"\Sigma", "Π": r"\Pi", "Δ": r"\Delta", "Φ": r"\Phi",
+               "Γ": r"\Gamma"}
 
 integer = pp.Word(pp.nums)
 molecule_Smult = integer.setResultsName("Smult")
diff --git a/tests/test_molecular_term_symbol.py b/tests/test_molecular_term_symbol.py
index 62cf5ab..b30489d 100644
--- a/tests/test_molecular_term_symbol.py
+++ b/tests/test_molecular_term_symbol.py
@@ -72,6 +72,18 @@ def test_greek_letter_conversion(self):
         self.assertEqual(str(m3), "3Σ+u")
         self.assertEqual(m3.html, "<sup>3</sup>Σ<sup>+</sup><sub>u</sub>")
 
+        m3 = MolecularTermSymbol("4GAMMA")
+        self.assertEqual(m3.Smult, 4)
+        self.assertEqual(m3.irrep, "Γ")
+        self.assertEqual(str(m3), "4Γ")
+        self.assertEqual(m3.html, "<sup>4</sup>Γ")
+
+        m3 = MolecularTermSymbol("2GAMMAg")
+        self.assertEqual(m3.Smult, 2)
+        self.assertEqual(m3.irrep, "Γg")
+        self.assertEqual(str(m3), "2Γg")
+        self.assertEqual(m3.html, "<sup>2</sup>Γ<sub>g</sub>")
+
         m3 = MolecularTermSymbol('A(1A")')
         self.assertEqual(str(m3), 'A(1A")')
         self.assertEqual(m3.html, 'A<sup>1</sup>A"')