def simulate_wave_for_computer(self, computer_name):
    last_wave = [computer_name]
    computer_waves = [computer_name]
    waves = []
    total = 0
    for wave in range(self.maxwaves):
        w_str = self.generate_wave_query_string()
        mysession = self.driver.session()
        result = mysession.run(w_str, last_wave=last_wave)
        for record in result:
            next_wave = record['next_wave']
            wave_size = len(next_wave)
            total += wave_size
            waves.append(str(wave_size))
            last_wave += next_wave
            if (wave_size == 0):
                mysession.close()
                return (total, waves)
        computer_waves.append(last_wave.copy())
        mysession.close()
    return (total, waves)
def simulate_wave_for_computer(self, computer_name):
    last_wave = [computer_name]
    computer_waves = [computer_name]
    waves = []
    total = 0
    for wave in range(self.maxwaves):
        w_str = self.generate_wave_query_string()
        mysession = self.driver.session()
        result = mysession.run(w_str, last_wave=last_wave)
        for record in result:
            next_wave = record['next_wave']
            wave_size = len(next_wave)
            total += wave_size
            waves.append(str(wave_size))
            last_wave += next_wave
            if (wave_size == 0):
                mysession.close()
                return (total, waves)
        computer_waves.append(last_wave.copy())
        mysession.close()
    return (total, waves)