Add unit tests for Equipment model