Additional unit tests for initial KMeans assignment, refactoring old unit tests to use assertRaise.