lionagi 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

lionagi/tests/test_core/graph/test_tree.py

@@ -1,75 +1,76 @@
-import unittest
-from lionagi.core.graph.tree import TreeNode, Tree
-
-class TestTreeNode(unittest.TestCase):
-    def setUp(self):
-        self.parent_node = TreeNode(id_="parent", content="Parent Node")
-        self.child_node1 = TreeNode(id_="child1", content="Child Node 1")
-        self.child_node2 = TreeNode(id_="child2", content="Child Node 2")
-
-    def test_relate_child(self):
-        self.parent_node.relate_child(self.child_node1)
-        self.assertIn("child1", self.parent_node.children)
-        self.assertEqual(self.child_node1.parent, self.parent_node)
-
-    def test_relate_children(self):
-        self.parent_node.relate_child([self.child_node1, self.child_node2])
-        self.assertIn("child1", self.parent_node.children)
-        self.assertIn("child2", self.parent_node.children)
-        self.assertEqual(self.child_node1.parent, self.parent_node)
-        self.assertEqual(self.child_node2.parent, self.parent_node)
-
-    def test_relate_parent(self):
-        self.child_node1.relate_parent(self.parent_node)
-        self.assertIn("child1", self.parent_node.children)
-        self.assertEqual(self.child_node1.parent, self.parent_node)
-
-    def test_unrelate_child(self):
-        self.parent_node.relate_child(self.child_node1)
-        self.parent_node.unrelate_child(self.child_node1)
-        self.assertNotIn("child1", self.parent_node.children)
-        self.assertIsNone(self.child_node1.parent)
-
-    def test_unrelate_parent(self):
-        self.child_node1.relate_parent(self.parent_node)
-        self.child_node1.unrelate_parent()
-        self.assertNotIn("child1", self.parent_node.children)
-        self.assertIsNone(self.child_node1.parent)
-
-class TestTree(unittest.TestCase):
-    def setUp(self):
-        self.tree = Tree()
-        self.root = TreeNode(id_="root", content="Root Node")
-        self.child_node1 = TreeNode(id_="child1", content="Child Node 1")
-        self.child_node2 = TreeNode(id_="child2", content="Child Node 2")
-        self.grandchild_node = TreeNode(id_="grandchild", content="Grandchild Node")
-
-    def test_add_node(self):
-        self.tree.add_node(self.root)
-        self.assertIn("root", self.tree.internal_nodes)
-
-    def test_relate_parent_child(self):
-        self.tree.relate_parent_child(self.root, [self.child_node1, self.child_node2])
-        self.assertIn("child1", self.root.children)
-        self.assertIn("child2", self.root.children)
-        self.assertEqual(self.tree.root, self.root)
-
-    def test_tree_structure(self):
-        # Build the tree
-        self.tree.relate_parent_child(self.root, [self.child_node1, self.child_node2])
-        self.tree.relate_parent_child(self.child_node1, self.grandchild_node)
-
-        # Check the tree structure
-        self.assertIn("grandchild", self.child_node1.children)
-        self.assertEqual(self.grandchild_node.parent, self.child_node1)
-        self.assertEqual(self.child_node1.parent, self.root)
-
-    def test_acyclic(self):
-        # Build the tree
-        self.tree.relate_parent_child(self.root, self.child_node1)
-        self.tree.relate_parent_child(self.child_node1, self.child_node2)
-        # Trees should always be acyclic
-        self.assertTrue(self.tree.acyclic)
-
-if __name__ == "__main__":
-    unittest.main()
+#TODO
+# import unittest
+# from lionagi.core.graph.tree import TreeNode, Tree
+#
+# class TestTreeNode(unittest.TestCase):
+#     def setUp(self):
+#         self.parent_node = TreeNode(id_="parent", content="Parent Node")
+#         self.child_node1 = TreeNode(id_="child1", content="Child Node 1")
+#         self.child_node2 = TreeNode(id_="child2", content="Child Node 2")
+#
+#     def test_relate_child(self):
+#         self.parent_node.relate_child(self.child_node1)
+#         self.assertIn("child1", self.parent_node.children)
+#         self.assertEqual(self.child_node1.parent, self.parent_node)
+#
+#     def test_relate_children(self):
+#         self.parent_node.relate_child([self.child_node1, self.child_node2])
+#         self.assertIn("child1", self.parent_node.children)
+#         self.assertIn("child2", self.parent_node.children)
+#         self.assertEqual(self.child_node1.parent, self.parent_node)
+#         self.assertEqual(self.child_node2.parent, self.parent_node)
+#
+#     def test_relate_parent(self):
+#         self.child_node1.relate_parent(self.parent_node)
+#         self.assertIn("child1", self.parent_node.children)
+#         self.assertEqual(self.child_node1.parent, self.parent_node)
+#
+#     def test_unrelate_child(self):
+#         self.parent_node.relate_child(self.child_node1)
+#         self.parent_node.unrelate_child(self.child_node1)
+#         self.assertNotIn("child1", self.parent_node.children)
+#         self.assertIsNone(self.child_node1.parent)
+#
+#     def test_unrelate_parent(self):
+#         self.child_node1.relate_parent(self.parent_node)
+#         self.child_node1.unrelate_parent()
+#         self.assertNotIn("child1", self.parent_node.children)
+#         self.assertIsNone(self.child_node1.parent)
+#
+# class TestTree(unittest.TestCase):
+#     def setUp(self):
+#         self.tree = Tree()
+#         self.root = TreeNode(id_="root", content="Root Node")
+#         self.child_node1 = TreeNode(id_="child1", content="Child Node 1")
+#         self.child_node2 = TreeNode(id_="child2", content="Child Node 2")
+#         self.grandchild_node = TreeNode(id_="grandchild", content="Grandchild Node")
+#
+#     def test_add_node(self):
+#         self.tree.add_node(self.root)
+#         self.assertIn("root", self.tree.internal_nodes)
+#
+#     def test_relate_parent_child(self):
+#         self.tree.relate_parent_child(self.root, [self.child_node1, self.child_node2])
+#         self.assertIn("child1", self.root.children)
+#         self.assertIn("child2", self.root.children)
+#         self.assertEqual(self.tree.root, self.root)
+#
+#     def test_tree_structure(self):
+#         # Build the tree
+#         self.tree.relate_parent_child(self.root, [self.child_node1, self.child_node2])
+#         self.tree.relate_parent_child(self.child_node1, self.grandchild_node)
+#
+#         # Check the tree structure
+#         self.assertIn("grandchild", self.child_node1.children)
+#         self.assertEqual(self.grandchild_node.parent, self.child_node1)
+#         self.assertEqual(self.child_node1.parent, self.root)
+#
+#     def test_acyclic(self):
+#         # Build the tree
+#         self.tree.relate_parent_child(self.root, self.child_node1)
+#         self.tree.relate_parent_child(self.child_node1, self.child_node2)
+#         # Trees should always be acyclic
+#         self.assertTrue(self.tree.acyclic)
+#
+# if __name__ == "__main__":
+#     unittest.main()

lionagi/tests/test_core/mail/test_mail.py

@@ -1,62 +1,98 @@
-from lionagi.core.generic.mail import *
-
-import unittest
-
-
-class TestMail(unittest.TestCase):
-    def setUp(self):
-        self.mail = Mail(
-            sender="node1",
-            recipient="node2",
-            category=MailPackageCategory.MESSAGES,
-            package="Hello, World!"
-        )
-
-    def test_mail_initialization(self):
-        """Test initialization of Mail objects."""
-        self.assertIsInstance(self.mail, BaseComponent)
-        self.assertEqual(self.mail.sender, "node1")
-        self.assertEqual(self.mail.recipient, "node2")
-        self.assertEqual(self.mail.category, MailPackageCategory.MESSAGES)
-        self.assertEqual(self.mail.package, "Hello, World!")
-
-    def test_mail_str(self):
-        """Test the string representation of Mail."""
-        expected_str = "Mail from node1 to node2 with category messages and package Hello, World!"
-        self.assertEqual(str(self.mail), expected_str)
-
-class TestMailBox(unittest.TestCase):
-    def setUp(self):
-        self.mailbox = MailBox()
-        self.mail1 = Mail(
-            sender="node1",
-            recipient="node3",
-            category="model",
-            package={"model": "Random Forest"}
-        )
-        self.mail2 = Mail(
-            sender="node2",
-            recipient="node3",
-            category=MailPackageCategory.SERVICE,
-            package={"service": "Prediction"}
-        )
-
-    def test_adding_mails(self):
-        """Test adding mails to MailBox."""
-        self.mailbox.pending_ins["node1"] = self.mail1
-        self.mailbox.pending_outs["node3"] = self.mail2
-
-        self.assertIn("node1", self.mailbox.pending_ins)
-        self.assertIn("node3", self.mailbox.pending_outs)
-        self.assertEqual(self.mailbox.pending_ins["node1"], self.mail1)
-        self.assertEqual(self.mailbox.pending_outs["node3"], self.mail2)
-
-    def test_mailbox_str(self):
-        """Test the string representation of MailBox."""
-        self.mailbox.pending_ins["node1"] = self.mail1
-        self.mailbox.pending_outs["node3"] = self.mail2
-        expected_str = "MailBox with 1 pending incoming mails and 1 pending outgoing mails."
-        self.assertEqual(str(self.mailbox), expected_str)
-
-if __name__ == "__main__":
-    unittest.main()
+import pytest
+
+from lionagi.core.mail import Mail
+from lionagi.core.mail.package import Package
+from datetime import datetime
+
+"""
+More to be added
+"""
+
+
+def trim_timestamp_to_day(timestamp_str):
+    dt = datetime.fromisoformat(timestamp_str)
+    return dt.strftime('%Y-%m-%d')
+
+
+class MockPackage(Package):
+    """A mock package class for testing purposes."""
+
+
+@pytest.fixture
+def mail():
+    """Fixture to create a Mail instance with a MockPackage."""
+    package = MockPackage()
+    mail_instance = Mail(timestamp=datetime.today().strftime('%Y-%m-%d'), package=package)
+    return mail_instance
+
+
+def test_mail_category(mail):
+    """Test the category property of Mail."""
+    assert mail.category is None
+
+
+
+
+
+# from lionagi.core.generic.mail import *
+#
+# import unittest
+#
+#
+# class TestMail(unittest.TestCase):
+#     def setUp(self):
+#         self.mail = Mail(
+#             sender="node1",
+#             recipient="node2",
+#             category=MailPackageCategory.MESSAGES,
+#             package="Hello, World!"
+#         )
+#
+#     def test_mail_initialization(self):
+#         """Test initialization of Mail objects."""
+#         self.assertIsInstance(self.mail, BaseComponent)
+#         self.assertEqual(self.mail.sender, "node1")
+#         self.assertEqual(self.mail.recipient, "node2")
+#         self.assertEqual(self.mail.category, MailPackageCategory.MESSAGES)
+#         self.assertEqual(self.mail.package, "Hello, World!")
+#
+#     def test_mail_str(self):
+#         """Test the string representation of Mail."""
+#         expected_str = "Mail from node1 to node2 with category messages and package Hello, World!"
+#         self.assertEqual(str(self.mail), expected_str)
+#
+# class TestMailBox(unittest.TestCase):
+#     def setUp(self):
+#         self.mailbox = MailBox()
+#         self.mail1 = Mail(
+#             sender="node1",
+#             recipient="node3",
+#             category="model",
+#             package={"model": "Random Forest"}
+#         )
+#         self.mail2 = Mail(
+#             sender="node2",
+#             recipient="node3",
+#             category=MailPackageCategory.SERVICE,
+#             package={"service": "Prediction"}
+#         )
+#
+#     def test_adding_mails(self):
+#         """Test adding mails to MailBox."""
+#         self.mailbox.pending_ins["node1"] = self.mail1
+#         self.mailbox.pending_outs["node3"] = self.mail2
+#
+#         self.assertIn("node1", self.mailbox.pending_ins)
+#         self.assertIn("node3", self.mailbox.pending_outs)
+#         self.assertEqual(self.mailbox.pending_ins["node1"], self.mail1)
+#         self.assertEqual(self.mailbox.pending_outs["node3"], self.mail2)
+#
+#     def test_mailbox_str(self):
+#         """Test the string representation of MailBox."""
+#         self.mailbox.pending_ins["node1"] = self.mail1
+#         self.mailbox.pending_outs["node3"] = self.mail2
+#         expected_str = "MailBox with 1 pending incoming mails and 1 pending outgoing mails."
+#         self.assertEqual(str(self.mailbox), expected_str)
+#
+# if __name__ == "__main__":
+#     unittest.main()