Function TestProcessTernary

query/math_test.go:971–1012

(t *testing.T)

}

func TestProcessTernary(t *testing.T) {
	tests := []struct {
		in  *mathTree
		out types.Val
	}{
		{in: &mathTree{
			Fn: "cond",
			Child: []*mathTree{
				{Val: createShardedMap(0, types.Val{Tid: types.BoolID, Value: true})},
				{Const: types.Val{Tid: types.IntID, Value: int64(1)}},
				{Const: types.Val{Tid: types.IntID, Value: int64(2)}},
			}},
			out: types.Val{Tid: types.IntID, Value: int64(1)},
		},
		{in: &mathTree{
			Fn: "cond",
			Child: []*mathTree{
				{Val: createShardedMap(0, types.Val{Tid: types.BoolID, Value: false})},
				{Const: types.Val{Tid: types.FloatID, Value: 1.0}},
				{Const: types.Val{Tid: types.FloatID, Value: 2.0}},
			}},
			out: types.Val{Tid: types.FloatID, Value: 2.0},
		},
		{in: &mathTree{
			Fn: "cond",
			Child: []*mathTree{
				{Val: createShardedMap(0, types.Val{Tid: types.BoolID, Value: false})},
				{Const: types.Val{Tid: types.BigFloatID, Value: *big.NewFloat(1.456)}},
				{Const: types.Val{Tid: types.BigFloatID, Value: *big.NewFloat(2.123)}},
			}},
			out: types.Val{Tid: types.BigFloatID, Value: *big.NewFloat(2.123)},
		},
	}
	for _, tc := range tests {
		t.Logf("Test: %s", tc.in.Fn)
		err := processTernary(tc.in)
		require.NoError(t, err)
		val, ok := tc.in.Val.Get(0)
		require.Equal(t, true, ok)
		require.EqualValues(t, tc.out, val)
	}
}