test_break.py

def was_break(self):
    driving_time = 0
    time_break = 0
    info_list = []
    short_break = False
    long_break = False
    max_time = 16200


    for record in self.records:
      if record.get_mode() == 'D':
          driving_time += record.get_value()

      if x.get_mode() == 'R'
          time_break += x.get_value()
          if 1800 <= time_break  and not long_break
            long_break = True
            time_break -= 1800
            max_time += 1800
          elif 900 <= time_break and not short_break
            short_break = True
            time_break -= 900
            max_time += 900
          
          #podmiana short na long
          if 1800 <= time_break + 900 and short_break and not long_break
            short_break = false
            long_break = true
            time_break -= 900
            max_time += 900

      if driving_time > max_time:
        info = "Break needed!"
        info_list.append(info)
    
   return info_list