提交 6c8cb50e authored 作者: Brandon T. Willard's avatar Brandon T. Willard

Apply isort to tests.sparse sub-package modules

上级 c1af419c
...@@ -2,20 +2,18 @@ import time ...@@ -2,20 +2,18 @@ import time
import pytest import pytest
pytest.importorskip("scipy", minversion="0.7.0") pytest.importorskip("scipy", minversion="0.7.0")
import numpy as np import numpy as np
from scipy.signal import convolve2d
import theano import theano
import theano.sparse import theano.sparse
from tests import unittest_tools as utt
from scipy.signal import convolve2d
from theano import function, tensor from theano import function, tensor
from theano.sparse.sandbox import sp from theano.sparse.sandbox import sp
from tests import unittest_tools as utt
class TestSP: class TestSP:
@pytest.mark.slow @pytest.mark.slow
......
...@@ -2,97 +2,93 @@ import time ...@@ -2,97 +2,93 @@ import time
import pytest import pytest
sp = pytest.importorskip("scipy", minversion="0.7.0")
import numpy as np
import theano sp = pytest.importorskip("scipy", minversion="0.7.0")
from itertools import product from itertools import product
import numpy as np
from packaging import version from packaging import version
import theano
from theano import tensor, sparse, compile, config, gof from tests import unittest_tools as utt
from tests.tensor.test_sharedvar import makeSharedTester
from theano.sparse.basic import ( from theano import compile, config, gof, sparse, tensor
_is_sparse,
_mtypes,
_is_dense_variable,
_is_sparse_variable,
)
from theano.sparse import ( from theano.sparse import (
as_sparse_variable,
as_sparse_or_tensor_variable,
CSR,
CSC, CSC,
CSM, CSM,
CSR,
AddSD,
AddSS,
AddSSData,
Cast,
ConstructSparseFromList,
CSMGrad,
CSMProperties, CSMProperties,
csm_properties,
DenseFromSparse, DenseFromSparse,
Diag,
Dot,
EnsureSortedIndices,
GetItemScalar,
HStack,
MulSD,
MulSS,
Neg,
Remove0,
SamplingDot,
SparseFromDense,
SparseType, SparseType,
CSMGrad, SquareDiagonal,
StructuredDot, StructuredDot,
StructuredDotGradCSC, StructuredDotGradCSC,
StructuredDotGradCSR, StructuredDotGradCSR,
AddSS,
AddSD,
MulSS,
MulSD,
Transpose, Transpose,
Neg, TrueDot,
Remove0, Usmm,
VStack,
add, add,
mul, add_s_s_data,
structured_dot, as_sparse_or_tensor_variable,
transpose, as_sparse_variable,
cast,
clean,
construct_sparse_from_list,
csc_from_dense, csc_from_dense,
csm_properties,
csr_from_dense, csr_from_dense,
dense_from_sparse, dense_from_sparse,
Dot,
Usmm,
sp_ones_like,
GetItemScalar,
SparseFromDense,
Cast,
cast,
HStack,
VStack,
AddSSData,
add_s_s_data,
structured_minimum,
structured_maximum,
structured_add,
mul_s_v,
structured_add_s_v,
SamplingDot,
sampling_dot,
Diag,
diag, diag,
SquareDiagonal,
square_diagonal,
EnsureSortedIndices,
ensure_sorted_indices, ensure_sorted_indices,
clean,
ConstructSparseFromList,
construct_sparse_from_list,
TrueDot,
true_dot,
le,
ge, ge,
gt, gt,
le,
lt, lt,
mul,
mul_s_v,
sampling_dot,
sp_ones_like,
square_diagonal,
structured_add,
structured_add_s_v,
structured_dot,
structured_maximum,
structured_minimum,
transpose,
true_dot,
)
from theano.sparse.basic import (
_is_dense_variable,
_is_sparse,
_is_sparse_variable,
_mtypes,
) )
from theano.sparse.opt import CSMGradC, StructuredDotCSC, UsmmCscDense
# Probability distributions are currently tested in test_sp2.py # Probability distributions are currently tested in test_sp2.py
# from theano.sparse import ( # from theano.sparse import (
# Poisson, poisson, Binomial, Multinomial, multinomial) # Poisson, poisson, Binomial, Multinomial, multinomial)
from theano.sparse.opt import StructuredDotCSC, UsmmCscDense, CSMGradC
from tests import unittest_tools as utt
from tests.tensor.test_sharedvar import makeSharedTester
def as_sparse_format(data, format): def as_sparse_format(data, format):
if format == "csc": if format == "csc":
......
import pytest import pytest
sp = pytest.importorskip("scipy", minversion="0.7.0") sp = pytest.importorskip("scipy", minversion="0.7.0")
import numpy as np import numpy as np
import theano import theano
from theano import sparse, config, tensor
from tests import unittest_tools as utt from tests import unittest_tools as utt
from tests.sparse.test_basic import random_lil from tests.sparse.test_basic import random_lil
from theano import config, sparse, tensor
def test_local_csm_properties_csm(): def test_local_csm_properties_csm():
......
import pytest import pytest
sp = pytest.importorskip("scipy", minversion="0.7.0") sp = pytest.importorskip("scipy", minversion="0.7.0")
import numpy as np import numpy as np
import theano import theano
from tests import unittest_tools as utt
from theano import config from tests.sparse.test_basic import as_sparse_format
from theano import tensor from theano import config, sparse, tensor
from theano import sparse
from theano.sparse.sandbox.sp2 import ( from theano.sparse.sandbox.sp2 import (
Poisson,
poisson,
Binomial, Binomial,
Multinomial, Multinomial,
Poisson,
multinomial, multinomial,
poisson,
) )
from tests import unittest_tools as utt
from tests.sparse.test_basic import as_sparse_format
class TestPoisson(utt.InferShapeTester): class TestPoisson(utt.InferShapeTester):
x = {} x = {}
......
import numpy as np
import pytest import pytest
import numpy as np
sp = pytest.importorskip("scipy", minversion="0.7.0") sp = pytest.importorskip("scipy", minversion="0.7.0")
import theano.sparse import theano.sparse
from theano.sparse.utils import hash_from_sparse
from tests.sparse.test_basic import as_sparse_format from tests.sparse.test_basic import as_sparse_format
from theano.sparse.utils import hash_from_sparse
def test_hash_from_sparse(): def test_hash_from_sparse():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论