Possibility to split tests on Class basis

I am currently using this plugin for splitting up my tests.
A lot of my pytests consist of multiple test cases which are depended on one another.
Is there a possibility to split the tests on class basis?
Example:
`Testclass A:
test_01:
test_02:
`
`Testclass B:
test_01:
test_02:
test_03:
`
`Testclass C:
test_01:
`
Splitting those in two would do something like this:
Batch A: A(test_01, test_02) C(test_01)
Batch B: B (test_01, test_02, test_03)

Currently I am getting something like this:
Batch A: A(test_01, test_02) B(test_01)
Batch B: B(test_02, test_03), C(test_01)

Maybe I am missing something. If this is not possible at the moment are you willing to accept a Pull Request?

Greetings,
Andreas

Start by reading the pytest-split implementation and its existing tests to find where pytest tests are assigned to batches. Check how collection items are grouped and weighted, then define completion as keeping each test class together while still producing the requested number of balanced batches.

Written by the indexing model from the issue text.