old_contents: # -*- coding: utf-8 -*-
  
  import os
  import pytest
  from koheron_tcp_client import KClient
  from lase.core import ZynqSSH
  from lase.drivers import Oscillo
  
  @pytest.mark.real
  class TestConnection:
        
      def test(self):
          current_path = os.getcwd()
          bitstreams_path = os.path.join(current_path,'bitstreams')
          host = os.getenv('HOST','192.168.1.100')
          client = KClient(host)        
          assert client.is_connected
          ssh = ZynqSSH(host, 'koheron')
          ssh.load_pl(os.path.join(bitstreams_path, 'oscillo.bit'))
          driver = Oscillo(client, current_mode='pwm')
          current = 30 #mA
          driver.set_laser_current(current)
          driver.start_laser()
          assert driver.get_laser_power() > 400
  
new_contents: # -*- coding: utf-8 -*-
  
  import os
  import pytest
  from koheron_tcp_client import KClient
  from ..core import ZynqSSH
  from ..drivers import Oscillo
  
  @pytest.mark.real
  class TestConnection:
        
      def test(self):
          current_path = os.getcwd()
          bitstreams_path = os.path.join(current_path,'bitstreams')
          host = os.getenv('HOST','192.168.1.100')
          client = KClient(host)        
          assert client.is_connected
          ssh = ZynqSSH(host, 'koheron')
          ssh.load_pl(os.path.join(bitstreams_path, 'oscillo.bit'))
          driver = Oscillo(client, current_mode='pwm')
          current = 30 #mA
          driver.set_laser_current(current)
          driver.start_laser()
          assert driver.get_laser_power() > 400
  
subject: edit tests