Simplify generate_input

pythonfuzz/corpus.py

@@ -27,10 +27,13 @@ def __init__(self, dirs=None, max_input_size=4096):
                     fname = os.path.join(path, i)
                     if os.path.isfile(fname):
                         self._add_file(fname)
+        if not self._inputs:
+            self.put(bytearray(0))
         self._seed_run_finished = not self._inputs
         self._seed_idx = 0
         self._save_corpus = dirs and os.path.isdir(dirs[0])
 
+
     def _add_file(self, path):
         with open(path, 'rb') as f:
             self._inputs.append(bytearray(f.read()))
@@ -92,13 +95,8 @@ def generate_input(self):
                 self._seed_run_finished = True
             return next_input
 
-        if len(self._inputs) == 0:
-            zero_test_case = bytearray(0)
-            self.put(zero_test_case)
-            return zero_test_case
-        else:
-            buf = self._inputs[self._rand(len(self._inputs))]
-            return self.mutate(buf)
+        buf = self._inputs[self._rand(len(self._inputs))]
+        return self.mutate(buf)
 
     def mutate(self, buf):
         res = buf[:]