Add visual regression tests to ensure it keeps working properly

[itaditya]

• Describe the Feature Request
  Add visual regression tests to ensure it keeps working properly using jest and jest-image-snapshot.

[itaditya]

If this is needed, I'd be happy to work on it.

[apollonian]

Oh, sorry, I missed this.
It would be extremely helpful to have regression tests!

[adithyaakrishna]

Hi @apollonian , If this is available, would be happy to work on this :)

[BhuwanChandra]

@apollonian I would like to work on this issue, please assign this to me!