Enhance generate_base_data() and Add Missing Test Cases for plot_energy()

[PiyushPanwarFST]

Current Limitations of generate_base_data():

1. It does not generate a prior group, which is essential for testing functions like plot_bf() that compare prior and posterior distributions.
2. It lacks an energy variable in the sample_stats group, which is required for testing energy-related plots like plot_energy().

[PiyushPanwarFST]

Heyy @aloctavodia here is the PR can you please review it #177

[aloctavodia]

I get notifications on every PR, commit and issue no need to ping me. Be patient. You can add [WIP] to the title of a PR to indicate when you are still working on PR and remove it, once is ready for review. Or mark a PR as a draft.
Also it seems you need to work on your git skills, usually there is no neex to close PRs and submit then again.

[PiyushPanwarFST]

Sorry for the mess. I'm actively improving and assure you it won’t happen again

[aloctavodia]

That's fine, mistakes help us learn.