Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 3 additions & 4 deletions pymc3/tests/test_diagnostics.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,7 @@
import unittest

import numpy as np
from numpy.testing import assert_allclose, assert_array_less

from .helpers import SeededTest
from ..model import Model
from ..step_methods import Slice, Metropolis, NUTS
from ..distributions import Normal
Expand All @@ -12,7 +11,7 @@
from .test_examples import build_disaster_model


class TestGelmanRubin(unittest.TestCase):
class TestGelmanRubin(SeededTest):
good_ratio = 1.1

def get_ptrace(self, n_samples):
Expand Down Expand Up @@ -85,7 +84,7 @@ def test_right_shape_scalar_one(self):
self.test_right_shape_python_float(shape=1, test_shape=(1,))


class TestDiagnostics(unittest.TestCase):
class TestDiagnostics(SeededTest):

def get_switchpoint(self, n_samples):
model = build_disaster_model()
Expand Down