Add Unit Tests

[dcchambers]

As more features are added/things are changed, need to add some unit tests to avoid regression bugs.