test: create unit and integration tests for all the plots

[SaaiVenkat]

Feature Request

Unit and Integration Testing

Description

The project lacks any kind of testing. To make sure we don't break because of third party changes and to account for all kinds od scenarios, introduce unit and integration testing.

Checklist

• I have searched for similar feature requests and confirmed that this is a new request.
• I have provided a clear and concise description of the feature.