Small fixes

1 files changed, 9 insertions, 2 deletions

diff --git a/2.1.py b/2.1.py
index 43896d6..63e4a99 100644
--- a/2.1.py
+++ b/2.1.py
@@ -1,4 +1,4 @@
-#!/usr/bin/pytnon2
+#!/usr/bin/python2
 
 ## You are given a list of "pegs" with their location (positive integers, asc order).
 ## these pegs can accept gears with radius > 1
@@ -9,7 +9,7 @@
 from fractions import gcd
 from functools import reduce
 
-# test
+# for tests
 from random import seed
 from random import random
 
@@ -45,6 +45,13 @@ def solution(pegs):
             return [-1, -1] 
         return [r1, 1]
 
+
+
+# test cases provided
+print(solution([4,30,50])) # [12, 1]
+print(solution([4,17,50])) # [-1, -1]
+
+# tests
 seed(1)
 
 def ran():