#include "driving.h"
#include <math.h>

_Complex double g_test(
  int kx,
  int ky
){
  //return sqrt(kx*kx*ky*ky)*exp(-(kx*kx+ky*ky));
  if(kx==2 && ky==-1){
    return 0.5+sqrt(3)/2*I;
  }
  else if(kx==-2 && ky==1){
     return 0.5-sqrt(3)/2*I;
  }
  return 0.;
}