Test: Flatten Test use pytest fixture or unitest set up

[HlaKarki]

Flatten Test should use pytest fixture or unitest set up

[trentonyo]

@HlaKarki
Bump

[HlaKarki]

I will be on it this week ✋🏼