refactoring

src/BatchcodeCompletion.py

@@ -6,6 +6,11 @@ class BatchcodeCompletion:
         self.ADR_by_Batchcode = ADR_by_Batchcode.sort_values(by = 'Adverse Reaction Reports', ascending = False)
 
     def completeBatchcode(self, partialBatchcode):
+        return self._getBatchcodeHavingMostADRs(self._filterBy(partialBatchcode))
+    
+    def _filterBy(self, partialBatchcode):
         smartRegexp = SmartRegexpFactory().createSmartRegexp(partialBatchcode)
-        filteredBbatchCodeTable = self.ADR_by_Batchcode[self.ADR_by_Batchcode.index.str.contains(smartRegexp, na=False, regex=True)]
-        return filteredBbatchCodeTable.index[0] if not filteredBbatchCodeTable.empty else None
+        return self.ADR_by_Batchcode[self.ADR_by_Batchcode.index.str.contains(smartRegexp, na = False, regex = True)]
+
+    def _getBatchcodeHavingMostADRs(self, ADR_by_Batchcode):
+        return ADR_by_Batchcode.index[0] if not ADR_by_Batchcode.empty else None