Added Local balancing unit tests