def test_zero3():
z = ZeroOperator(shapein=3, shapeout=6)
o = Operator(flags='square')
zo = z*o
assert_is_instance(zo, ZeroOperator)
assert_equal(zo.shapein, z.shapein, 'in')
assert_equal(zo.shapeout, z.shapeout, 'out')
def test_zero6():
@flags.linear
class Op(Operator):
def direct(self, input, output):
output[:] = np.concatenate([input, 2*input])
def transpose(self, input, output):
output[:] = input[0:output.size]
def reshapein(self, shapein):
return (2 * shapein[0],)
def reshapeout(self, shapeout):
return (shapeout[0] // 2,)
z = ZeroOperator(flags='square')
o = Op()
od = o.todense(shapein=4)
zo = z * o
zod_ref = np.dot(np.zeros((8, 8)), od)
assert_same((z * o).todense(shapein=4), zod_ref)
oz = o * z
ozd_ref = np.dot(od, np.zeros((4, 4)))
assert_same((o * z).todense(shapein=4), ozd_ref)
assert_same(zo.T.todense(shapein=8), zod_ref.T)
assert_same(oz.T.todense(shapein=8), ozd_ref.T)
def test_zero2():
z = ZeroOperator(shapein=3, shapeout=6)
o = Operator()
zo = z(o)
assert_is_instance(zo, ZeroOperator)
assert_is_none(zo.shapein, 'in')
assert_equal(zo.shapeout, z.shapeout, 'out')
def test_zero1():
z = ZeroOperator()
o = Operator(shapein=3, shapeout=6)
zo = z(o)
assert_is_instance(zo, ZeroOperator)
assert_equal(zo.shapein, o.shapein)
assert_is_none(zo.shapeout)
def test_zero5():
z = ZeroOperator()
o = Operator(shapein=3, shapeout=6, flags='linear')
zo = z*o
oz = o*z
assert_is_instance(zo, ZeroOperator, 'zo')
assert_equal(zo.shapein, o.shapein, 'zo in')
assert_is_none(zo.shapeout, 'zo out')
assert_is_instance(oz, ZeroOperator, 'oz')
assert_is_none(oz.shapein, 'oz, in')
assert_equal(oz.shapeout, o.shapeout, 'oz, out')
def test_zero7():
z = ZeroOperator()
assert_equal(z*z, z)
def test_zero4():
z = ZeroOperator()
o = Operator(flags='linear')
assert_is_instance(z*o, ZeroOperator)
assert_is_instance(o*z, ZeroOperator)
from __future__ import division
import numpy as np
from numpy.testing import assert_equal
from pyoperators import (
CompositionOperator, ConstantOperator, Operator, ZeroOperator, flags,
rule_manager, O)
from pyoperators.utils import ndarraywrap
from pyoperators.utils.testing import (
assert_is, assert_is_instance, assert_is_none, assert_same, assert_is_type)
from .common import OPS, ndarray2, attr2
op = Operator()
ops = [_() for _ in OPS] + [_(flags={'linear': False}) for _ in OPS]
zeros_left = (
ZeroOperator(classout=ndarray2, attrout=attr2),
ZeroOperator(shapein=4, classout=ndarray2, attrout=attr2))
zeros_right = (
ZeroOperator(classout=ndarray2, attrout=attr2),
ZeroOperator(classout=ndarray2, attrout=attr2, flags='square'),
ZeroOperator(shapein=3, classout=ndarray2, attrout=attr2),
ZeroOperator(shapein=3, shapeout=3, classout=ndarray2, attrout=attr2))
def test_zero1():
z = ZeroOperator()
o = Operator(shapein=3, shapeout=6)
zo = z(o)
assert_is_instance(zo, ZeroOperator)
assert_equal(zo.shapein, o.shapein)
assert_is_none(zo.shapeout)