blob: 5e00b1e127fe6572c9cd028943deb2783a48a2cb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
"""
Pytest configuration and fixtures for the Numpy test suite.
"""
import pytest
import numpy
import matplotlib
import doctest
matplotlib.use('agg', force=True)
# Ignore matplotlib output such as `<matplotlib.image.AxesImage at
# 0x7f956908c280>`. doctest monkeypatching inspired by
# https://github.com/wooyek/pytest-doctest-ellipsis-markers (MIT license)
OutputChecker = doctest.OutputChecker
empty_line_markers = ['<matplotlib.', '<mpl_toolkits.mplot3d.']
class SkipMatplotlibOutputChecker(doctest.OutputChecker):
def check_output(self, want, got, optionflags):
for marker in empty_line_markers:
if marker in got:
got = ''
break
return OutputChecker.check_output(self, want, got, optionflags)
doctest.OutputChecker = SkipMatplotlibOutputChecker
@pytest.fixture(autouse=True)
def add_np(doctest_namespace):
numpy.random.seed(1)
doctest_namespace['np'] = numpy
|
