Added aastex.Affiliation class.

aastex/_latex.py

@@ -3,6 +3,7 @@
 
 __all__ = [
     "Title",
+    "Affiliation",
 ]
 
 
@@ -12,3 +13,14 @@ class Title(pylatex.base_classes.LatexObject):
 
     def dumps(self):
         return pylatex.Command("title", self.name).dumps()
+
+
+@dataclasses.dataclass
+class Affiliation(pylatex.base_classes.LatexObject):
+    """Organization that an author is associated with"""
+
+    name: str
+    """human-readable name of the organization"""
+
+    def dumps(self):
+        return pylatex.Command('affiliation', self.name).dumps()

aastex/_tests/test_latex.py

@@ -14,3 +14,17 @@ def test_name(self, a: aastex.Title):
 
     def test_dumps(self, a: aastex.Title):
         assert isinstance(a.dumps(), str)
+
+
+@pytest.mark.parametrize(
+    argnames="a",
+    argvalues=[
+        aastex.Affiliation("Fancy University"),
+    ],
+)
+class TestAffiliation:
+    def test_name(self, a: aastex.Affiliation):
+        assert isinstance(a.name, str)
+
+    def test_dumps(self, a: aastex.Affiliation):
+        assert isinstance(a.dumps(), str)