copy object to itself about pleque HOT 6 CLOSED

class B(object):

    def __init__(self, name):
        self.name = name


def B_factory(a):
    if isinstance(a, str):
        return B(a)
    elif isinstance(a, B):
        return a
    else:
        raise TypeError('Unknown name argument')


def test_B_factory():
    a = B_factory('foo')
    b = B_factory(a)
    assert isinstance(a, B)
    assert a is b
    assert a.name is b.name

class A(object):                # necessary for new-style class in PY2

    def __new__(cls, name):  # actual instance creation
        if isinstance(name, str):  # note PY2 incompatibility
            return super(A, cls).__new__(cls)  # calls object(), will call __init__
        elif isinstance(name, cls):
            return name  # would call __init__ again, need short-circuit in REF1
        else:
            raise TypeError('Unknown name argument')

    def __init__(self, name):  # initializes an instance
        if hasattr(self, 'name'):
            return              # already initialized REF1
        self.name = name


def test_A():
    a = A('foo')
    b = A(a)
    assert a is b
    assert a.name is b.name

a = A('foo')
b = A(a)
assert b == A(a) and b is not A  # b is a copy of a

a = A('foo')
b = A(a)
assert b is A(a)  # b is identical to a

class A:

    def __init__(self, name):
        self.name = name

    @classmethod
    def maybe_create_new_instance(cls, name_or_instance):
        if isinstance(name_or_instance, cls):
            return name_or_instance
        return cls(name_or_instance)


a = A('foo')
assert a is A.maybe_create_new_instance(a)  # b is identical to a
assert a is not A.maybe_create_new_instance('foo')  # creates a new instance

from pleque.core import Coordinates
from test.testing_utils import load_testing_equilibrium

eq = load_testing_equilibrium()

# 2d coordinates:
coord = Coordinates(eq, 1, 2)
coord = Coordinates(eq, R=1, Z=2)
coord = Coordinates(eq, [1, 2, 3, 4], [3, 5, 6, 2])
coord = Coordinates(eq, ([1, 3], [3, 5], [5, 3]))
coord = Coordinates(eq, np.array(([1, 3], [3, 5], [5, 3])), coord_type=['Z', 'R'])
coord = Coordinates(eq, R=[1, 2, 3, 4], Z=[3, 5, 6, 2])
coord = Coordinates(eq, Z=[3, 5, 6, 2], R=[1, 2, 3, 4])

# 1d coordinates(psi_n):
coord = Coordinates(eq, 0.5)
coord = Coordinates(eq, psi_n=0.5)
coord = Coordinates(eq, np.linspace(0, 1, 6), coord_type='rho')
coord = Coordinates(eq, psi=[0.4, 0.35, 0.3, 0.2, 0.15])
coord = Coordinates(eq, rho=[0, 0.2, 0.4, 0.6, 0.8, 1])
coord = Coordinates(eq, np.linspace(0, 1, 6), coord_type=('rho',))

# what I want:
new_coord = Coordinates(coord)
# not implemented yet
assert all(new_coord == coord)
# probably...
assert new_coord is not coord

a = np.linspace(0,10,10)
b = np.array(a)

assert all(a == b)
assert a is not b