Add Assert support in tests