import pytest

@pytest.fixture(scope="module")
def dists_scores():
	from ngboost.distns import Bernoulli, k_categorical, Normal, LogNormal, Exponential
	from ngboost.scores import LogScore, CRPScore
	return {
		Bernoulli:[LogScore],
		k_categorical(4): [LogScore], 
		Normal: [LogScore, CRPScore], 
		LogNormal: [LogScore, CRPScore], 
		Exponential: [LogScore, CRPScore]
		}

@pytest.fixture(scope="module")
def learners():
	from sklearn.tree import DecisionTreeRegressor
	return [
		DecisionTreeRegressor(criterion='friedman_mse', max_depth=5),
		DecisionTreeRegressor(criterion='friedman_mse', max_depth=3)
	]

class TestRegDistns():
	@pytest.fixture(scope="class")
	def reg_dists(self, dists_scores):
		from ngboost.distns import RegressionDistn
		return [dist for dist in dists_scores.keys() if isinstance(dist, RegressionDistn)]

	@pytest.fixture(scope="class")
	def reg_data(self):
		from sklearn.datasets import load_boston
		from sklearn.model_selection import train_test_split
		X, Y = load_boston(True)
		return train_test_split(X, Y, test_size=0.2)

	def test_which_reg_dists(self, reg_dists):
		from ngboost.distns import Normal, LogNormal, Exponential
		assert set([Normal, LogNormal, Exponential]) == set(reg_dists)

	def test_dists(self, dists_scores, learners, reg_dists, reg_data):
		from ngboost import NGBRegressor
		X_reg_train, X_reg_test, Y_reg_train, Y_reg_test = reg_data
		for Dist in reg_dists:
			for Score in dists_scores[Dist]:
				for Learner in learners:
					ngb = NGBRegressor(Dist=Dist, Score=Score, Base=Learner, verbose=False)
					ngb.fit(X_reg_train, Y_reg_train)
					y_pred = ngb.predict(X_reg_test)
					y_dist = ngb.pred_dist(X_reg_test)

# test slicing and ._params
@@ -610,32 +610,8 @@
 %09%5D%0A%0A
-class TestRegDistns():%0A%09
 @pyt
@@ -629,33 +629,32 @@
 (scope=%22class%22)%0A
-%09
 def reg_dists(se
@@ -647,30 +647,24 @@
 f reg_dists(
-self, 
 dists_scores
@@ -662,25 +662,24 @@
 ts_scores):%0A
-%09
 %09from ngboos
@@ -710,17 +710,16 @@
 onDistn%0A
-%09
 %09return 
@@ -795,17 +795,16 @@
 istn)%5D%0A%0A
-%09
 @pytest.
@@ -826,17 +826,16 @@
 class%22)%0A
-%09
 def reg_
@@ -843,16 +843,11 @@
 ata(
-self
 ):%0A
-%09
 %09fro
@@ -884,17 +884,16 @@
 _boston%0A
-%09
 %09from sk
@@ -938,17 +938,16 @@
 t_split%0A
-%09
 %09X, Y = 
@@ -964,17 +964,16 @@
 n(True)%0A
-%09
 %09return 
@@ -1011,16 +1011,40 @@
 e=0.2)%0A%0A
+class TestRegDistns():%0A%0A
 %09def tes

Update test_distns.py