import unittest

from A059861 import brute_force, r_times_r_plus_two


class Test(unittest.TestCase):

    # See https://oeis.org/A059861
    def test_brute_force(self):
        self.assertEqual(1, brute_force(1))
        self.assertEqual(1, brute_force(2))
        self.assertEqual(3, brute_force(3))
        self.assertEqual(15, brute_force(4))
        self.assertEqual(135, brute_force(5))
        self.assertEqual(1485, brute_force(6))

    def test_r_times_r_plus_two(self):
        self.assertEqual(1, r_times_r_plus_two(1))
        self.assertEqual(1, r_times_r_plus_two(2))
        self.assertEqual(3, r_times_r_plus_two(3))
        self.assertEqual(15, r_times_r_plus_two(4))
        self.assertEqual(135, r_times_r_plus_two(5))
        self.assertEqual(1485, r_times_r_plus_two(6))


if __name__ == '__main__':
    unittest.main()

import unittest

from A059861 import brute_force, r_times_r_plus_two


class Test(unittest.TestCase):

    # See https://oeis.org/A059861
    def test_brute_force(self):
        self.assertEqual(1, brute_force(1))
        self.assertEqual(1, brute_force(2))
        self.assertEqual(3, brute_force(3))
        self.assertEqual(15, brute_force(4))
        self.assertEqual(135, brute_force(5))
        self.assertEqual(1485, brute_force(6))

    def test_r_times_r_plus_two(self):
        self.assertEqual(1, r_times_r_plus_two(1))
        self.assertEqual(1, r_times_r_plus_two(2))
        self.assertEqual(3, r_times_r_plus_two(3))
        self.assertEqual(15, r_times_r_plus_two(4))
        self.assertEqual(135, r_times_r_plus_two(5))
        self.assertEqual(1485, r_times_r_plus_two(6))


if __name__ == '__main__':
    unittest.main()

# See https://oeis.org/A059861