# Path: tests/context.py

# 
# Path: tests/context.py



import numpy as np
import copy
import pdb
import os
import time
import matplotlib
import matplotlib.pylab as plt
from .context import pep, vfe
from .context import pep_tmp as pep
from test_utils import check_grad
matplotlib.use('Agg')


np.random.seed(42)

def test_gpr_pep_gaussian():
    N_train = 20
    M = 5
    D = 2
    Q = 3
    y_train = np.random.randn(N_train, Q)
    x_train = np.random.randn(N_train, D)

# Path: tests/context.py

# 
# Path: tests/context.py



# Path: tests/test_grads_pep.py
import numpy as np
import copy
import pdb
import os
import time
import matplotlib
import matplotlib.pylab as plt
from .context import pep, vfe
from .context import pep_tmp as pep
from test_utils import check_grad

matplotlib.use('Agg')


np.random.seed(42)

def test_gpr_pep_gaussian():
    N_train = 20
    M = 5
    D = 2
    Q = 3
    y_train = np.random.randn(N_train, Q)
    x_train = np.random.randn(N_train, D)

    model = pep.SGPR_rank_one(x_train, y_train, M, lik='Gaussian')