add some tests for basic balancing functionality