| import unittest | |
| from chain import rag_chain | |
| class TestRAGChain(unittest.TestCase): | |
| def test_rag_chain(self): | |
| ans = rag_chain.invoke("Who are Meta's 'Directors' (i.e., members of the Board of Directors)?") | |
| self.assertTrue(ans != "Peggy Alford, Marc L. Andreessen, Andrew W. Houston, Nancy Killefer, Robert M. Kimmitt, Sheryl K. Sandberg, Tracey T. Travis, Tony Xu") | |
| if __name__ == '__main__': | |
| unittest.main() |