Diophantine Equation

kyu_5/diophantine_equation/__init__.py

@@ -0,0 +1 @@
+"""Diophantine Equation."""

kyu_5/diophantine_equation/solution.py

@@ -1,14 +1,19 @@
 """
-Solution for -> Diophantine Equation
+Solution for -> Diophantine Equation.
+
 Created by Egor Kostan.
 GitHub: https://github.com/ikostan
 """
 
 
 def sol_equa(n: int) -> list:
     """
+    Diophantine Equation solution.
+
     Finds all integers x, y (x >= 0, y >= 0)
     solutions of a diophantine equation of the form x2 - 4 * y2 = n
+    :param n: int
+    :return: list
     """
     result: list = []
 

kyu_5/diophantine_equation/test_solution.py

@@ -1,5 +1,6 @@
 """
-Test for -> Diophantine Equation
+Test for -> Diophantine Equation.
+
 Created by Egor Kostan.
 GitHub: https://github.com/ikostan
 """
@@ -27,13 +28,12 @@
     name='Source/Kata')
 @pytest.mark.skip(reason="The solution is not ready")
 class SolutionTestCase(unittest.TestCase):
-    """
-    Testing sol_equa function
-    """
+    """Testing sol_equa function."""
 
     def test_solution_basic(self):
         """
-        Testing using basic test data
+        Testing using basic test data.
+
         :return:
         """
         self.assertEqual(sol_equa(5), [[3, 1]])
@@ -45,7 +45,8 @@ def test_solution_basic(self):
 
     def test_solution_medium(self):
         """
-        Testing using medium test data
+        Testing using medium test data.
+
         :return:
         """
         self.assertEqual(sol_equa(9001), [[4501, 2250]])
@@ -69,7 +70,8 @@ def test_solution_medium(self):
 
     def test_solution_big(self):
         """
-        Testing using big test data
+        Testing using big test data.
+
         :return:
         """
         self.assertEqual(sol_equa(900000), [[112502, 56249],
@@ -107,7 +109,8 @@ def test_solution_big(self):
 
     def test_solution_empty(self):
         """
-        Testing using empty test data
+        Testing using empty test data.
+
         :return:
         """
         self.assertEqual(sol_equa(90002), [])