提交 12b74a54 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix reduction test to actually iterate through all the dtypes.

上级 9bddfc4f
...@@ -784,12 +784,10 @@ class T_reduce_dtype(unittest.TestCase): ...@@ -784,12 +784,10 @@ class T_reduce_dtype(unittest.TestCase):
op = CAReduce op = CAReduce
axes = [None, 0, 1, [], [0], [1], [0, 1]] axes = [None, 0, 1, [], [0], [1], [0, 1]]
methods = ['sum', 'prod'] methods = ['sum', 'prod']
dtypes = imap(str, theano.scalar.all_types) dtypes = list(imap(str, theano.scalar.all_types))
# Test the default dtype of a method().
def test_reduce_default_dtype(self): def test_reduce_default_dtype(self):
"""
Test the default dtype of a method().
"""
# We try multiple axis combinations even though axis should not matter. # We try multiple axis combinations even though axis should not matter.
for method in self.methods: for method in self.methods:
for idx, dtype in enumerate(self.dtypes): for idx, dtype in enumerate(self.dtypes):
...@@ -840,9 +838,7 @@ class T_reduce_dtype(unittest.TestCase): ...@@ -840,9 +838,7 @@ class T_reduce_dtype(unittest.TestCase):
@attr('slow') @attr('slow')
def test_reduce_custom_dtype(self): def test_reduce_custom_dtype(self):
""" # Test the ability to provide your own output dtype for a reduce.
Test the ability to provide your own output dtype for a reduce.
"""
# We try multiple axis combinations even though axis should not matter. # We try multiple axis combinations even though axis should not matter.
idx = 0 idx = 0
for method in self.methods: for method in self.methods:
...@@ -875,9 +871,7 @@ class T_reduce_dtype(unittest.TestCase): ...@@ -875,9 +871,7 @@ class T_reduce_dtype(unittest.TestCase):
idx += 1 idx += 1
def test_reduce_custom_acc_dtype(self): def test_reduce_custom_acc_dtype(self):
""" # Test the ability to provide your own accumulator dtype for a reduce.
Test the ability to provide your own accumulator dtype for a reduce.
"""
# We try multiple axis combinations even though axis should not matter. # We try multiple axis combinations even though axis should not matter.
idx = 0 idx = 0
for method in self.methods: for method in self.methods:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论