class TestRigidGraphs(unittest.TestCase):
    def setUp(self):
        self.uut = RigidGraphs()

    def test_rigid(self):
        self.assertEqual(self.uut.rigid(2, 3), 19)
        self.assertEqual(self.uut.rigid(5, 5), 23679901)

    def test_rigidSum(self):
        self.assertEqual(self.uut.sum(5), 25021721)
 def setUp(self):
     self.uut = RigidGraphs()