Add unit tests for Display library