[Vega] Add unit test