[REFACTOR] [Hacker Rank] Interview Preparation Kit: Miscellaneous: Ice Cream Parlor. Brute force solution added.

Author: Gonzalo Diaz

src/hackerrank/interview_preparation_kit/search/ctci_ice_cream_parlor.border_testcases.json

@@ -0,0 +1,12 @@
+[
+  {
+    "title": "Sample Test Case 0",
+    "tests": [
+      {
+        "costs": [1, 4, 5, 3, 2],
+        "money": 90,
+        "expected": null
+      }
+    ]
+  }
+]

src/hackerrank/interview_preparation_kit/search/ctci_ice_cream_parlor.testcases.json

@@ -0,0 +1,42 @@
+[
+  {
+    "title": "Sample Test Case 0",
+    "tests": [
+      {
+        "costs": [1, 4, 5, 3, 2],
+        "money": 4,
+        "expected": [1, 4]
+      },
+      {
+        "costs": [2, 2, 4, 3],
+        "money": 4,
+        "expected": [1, 2]
+      }
+    ]
+  },
+  {
+    "title": "Sample Test Case 1",
+    "tests": [
+      {
+        "costs": [1, 2, 3, 5, 6],
+        "money": 5,
+        "expected": [2, 3]
+      }
+    ]
+  },
+  {
+    "title": "Sample Test Case 2",
+    "tests": [
+      {
+        "costs": [4, 3, 2, 5, 7],
+        "money": 8,
+        "expected": [2, 4]
+      },
+      {
+        "costs": [7, 2, 5, 4, 11],
+        "money": 12,
+        "expected": [1, 3]
+      }
+    ]
+  }
+]

src/hackerrank/interview_preparation_kit/search/ctci_ice_cream_parlor_bruteforce.py

@@ -0,0 +1,12 @@
+def what_flavors_brute_force(cost: list[int], money: int) -> list[int] | None:
+
+    for i, x in enumerate(cost):
+
+        budget = money - x
+
+        for j in range(i + 1, len(cost)):
+            if budget - cost[j] == 0:
+                print(f'{i + 1} {j + 1}')
+                return [i + 1, j + 1]
+
+    return []

src/hackerrank/interview_preparation_kit/search/ctci_ice_cream_parlor_bruteforce_test.py

@@ -0,0 +1,41 @@
+import unittest
+import json
+from pathlib import Path
+import os
+
+from .ctci_ice_cream_parlor_bruteforce import what_flavors_brute_force
+
+BRUTEFORCE = os.getenv('BRUTEFORCE')
+BRUTEFORCE = BRUTEFORCE.upper() == "TRUE" if BRUTEFORCE is not None else False
+
+
+FILE_PATH = str(Path(__file__).resolve().parent)
+
+with open(
+        FILE_PATH +
+        '/ctci_ice_cream_parlor.testcases.json',
+        encoding="utf-8"
+        ) as file1:
+    TEST_CASES = json.load(file1)
+
+with open(
+        FILE_PATH +
+        '/ctci_ice_cream_parlor.border_testcases.json',
+        encoding="utf-8"
+        ) as file2:
+    TEST_CASES_BORDER_CASES = json.load(file2)
+
+
+class TestIceCreamParlor(unittest.TestCase):
+
+    @unittest.skipIf(not BRUTEFORCE, "skipping due a is a large BRUTEFORCE test")
+    def test_what_flavors_brute_force(self):
+
+        for _, testset in enumerate(TEST_CASES):
+
+            for _, _tt in enumerate(testset['tests']):
+
+                self.assertEqual(
+                    what_flavors_brute_force(_tt['costs'], _tt['money']), _tt['expected'],
+                    f"{_} | what_flavors_brute_force({_tt['costs']}, {_tt['money']}) "
+                    f"=> must be {_tt['expected']}")

src/hackerrank/interview_preparation_kit/search/ctci_ice_cream_parlor_test.py

@@ -1,67 +1,25 @@
 import unittest
+import json
+from pathlib import Path
 
 from .ctci_ice_cream_parlor import what_flavors
 
 
-TEST_CASES = [
-    {
-        'title': 'Sample Test Case 0',
-        'tests':
-        [
-            {
-                'costs': [1, 4, 5, 3, 2],
-                'money': 4,
-                'answer': [1, 4]
-            },
-            {
-                'costs': [2, 2, 4, 3],
-                'money': 4,
-                'answer': [1, 2]
-            }
-        ]
-    },
-    {
-        'title': 'Sample Test Case 1',
-        'tests':
-        [
-            {
-                'costs': [1, 2, 3, 5, 6],
-                'money': 5,
-                'answer': [2, 3]
-            }
-        ]
-    },
-    {
-        'title': 'Sample Test Case 2',
-        'tests':
-        [
-            {
-                'costs': [4, 3, 2, 5, 7],
-                'money': 8,
-                'answer': [2, 4]
-            },
-            {
-                'costs': [7, 2, 5, 4, 11],
-                'money': 12,
-                'answer': [1, 3]
-            }
-        ]
-    }
-]
+FILE_PATH = str(Path(__file__).resolve().parent)
 
-TEST_CASES_BORDER_CASES = [
-    {
-        'title': 'Sample Test Case 0',
-        'tests':
-        [
-            {
-                'costs': [1, 4, 5, 3, 2],
-                'money': 90,
-                'answer': None
-            }
-        ]
-    }
-]
+with open(
+        FILE_PATH +
+        '/ctci_ice_cream_parlor.testcases.json',
+        encoding="utf-8"
+        ) as file1:
+    TEST_CASES = json.load(file1)
+
+with open(
+        FILE_PATH +
+        '/ctci_ice_cream_parlor.border_testcases.json',
+        encoding="utf-8"
+        ) as file2:
+    TEST_CASES_BORDER_CASES = json.load(file2)
 
 
 class TestIceCreamParlor(unittest.TestCase):
@@ -73,9 +31,9 @@ def test_what_flavors(self):
             for _, _tt in enumerate(testset['tests']):
 
                 self.assertEqual(
-                    what_flavors(_tt['costs'], _tt['money']), _tt['answer'],
+                    what_flavors(_tt['costs'], _tt['money']), _tt['expected'],
                     f"{_} | what_flavors({_tt['costs']}, {_tt['money']}) must be "
-                    f"=> {_tt['answer']}")
+                    f"=> {_tt['expected']}")
 
     def test_what_flavors_border_case(self):
 
@@ -84,6 +42,6 @@ def test_what_flavors_border_case(self):
             for _, _tt in enumerate(testset['tests']):
 
                 self.assertEqual(
-                    what_flavors(_tt['costs'], _tt['money']), _tt['answer'],
+                    what_flavors(_tt['costs'], _tt['money']), _tt['expected'],
                     f"{_} | what_flavors({_tt['costs']}, {_tt['money']}) must be "
-                    f"=> {_tt['answer']}")
+                    f"=> {_tt['expected']}")