[assign] Add tests for variants