Maybe you could add assertions on the count of each of the 5 possibilities to check that count / n_estimators is approximately 0.17 (or 0.33 for the most compact). Our would it require much more than 1000 trees and thus too slow to run to get a stable test robust to rng seed changes?