implemented 'sample' method in 'Localization'

squaresLab · Apr 1, 2018 · 8ed1376 · 8ed1376
1 parent 18642e3
commit 8ed1376
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bugzoo/localization/localization.py b/bugzoo/localization/localization.py
@@ -70,7 +70,12 @@ def sample(self) -> FileLine:
             ValueError: if there are no suspicious lines within this fault
                 localization.
         """
-        raise NotImplementedError
+        lines = list(self.__normalized.keys())
+        dist = [self.__normalized[line] for line in lines]
+        try:
+            return numpy.random.choice(lines, p=dist)
+        except ValueError:
+            raise ValueError
 
     def without_line(self,
                      line: FileLine,