diff --git a/src/codewars/kata_force_of_gravity.py b/src/codewars/kata_force_of_gravity.py index b576b00..3780d5d 100644 --- a/src/codewars/kata_force_of_gravity.py +++ b/src/codewars/kata_force_of_gravity.py @@ -1,5 +1,20 @@ # https://www.codewars.com/kata/5b609ebc8f47bd595e000627/ +units = { + "kg": 1, + "g": 1e-3, + "mg": 1e-6, + "μg": 1e-9, + "lb": 0.453592, + "m": 1, + "cm": 1e-2, + "mm": 1e-3, + "μm": 1e-6, + "ft": 0.3048, + "G": 6.67e-11, +} + def solution(arr_val, arr_unit): - pass + m1, m2, r = (arr_val[i] * units[arr_unit[i]] for i in range(3)) + return units["G"] * m1 * m2 / r**2 diff --git a/tests/codewars/test_kata_force_of_gravity.py b/tests/codewars/test_kata_force_of_gravity.py index 5ab8e39..84b7977 100644 --- a/tests/codewars/test_kata_force_of_gravity.py +++ b/tests/codewars/test_kata_force_of_gravity.py @@ -1,6 +1,7 @@ from src.codewars.kata_force_of_gravity import solution -def basic_test_cases(): + +def test_a(): assert solution([1000, 1000, 100], ["g", "kg", "m"]) == 6.67e-12 - assert solution([1000, 1000, 100], ["kg", "kg", "m"]) == 6.667e-9 - assert solution([1000, 1000, 100], ["kg", "kg", "cm"]) == 0.0000667 \ No newline at end of file + assert solution([1000, 1000, 100], ["kg", "kg", "m"]) == 6.6699999999999995e-09 + assert solution([1000, 1000, 100], ["kg", "kg", "cm"]) == 0.0000667