add test for 2 and 3 states #491 #492

Closed fnrizzi closed 1 year ago