feat: Add unit tests for the plot module (#153)

feat: Add unit tests for the plot module (#153) #366