Adding tests for field id keyed input data predictions