Example #1
0
def test_astype():
    rn = Rn(3, weight=1.5)
    cn = Cn(3, weight=1.5)
    rn_s = Rn(3, weight=1.5, dtype='float32')
    cn_s = Cn(3, weight=1.5, dtype='complex64')

    # Real
    assert rn.astype('float32') == rn_s
    assert rn.astype('float64') is rn
    assert rn._real_space is rn
    assert rn.astype('complex64') == cn_s
    assert rn.astype('complex128') == cn
    assert rn._complex_space == cn

    # Complex
    assert cn.astype('complex64') == cn_s
    assert cn.astype('complex128') is cn
    assert cn._complex_space is cn
    assert cn.astype('float32') == rn_s
    assert cn.astype('float64') == rn
    assert cn._complex_space is cn