Fixed test_simulation so the file directories are valid for all computers.

Fixed test for compute_gen_imbalance.

Added new argument to compute_gen_imbalance to allow user to choose what kind of value to return.