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()