def test_copy(self):
X = XLayer(
name='Elon',
type=['Machine'],
data=[np.array([1, 2, 3], dtype=np.float32)],
attrs={
'a': [1, 2],
'b': -1
}
)
X_copy = X.copy()
assert X_copy.name == 'Elon'
assert X_copy.type == ['Machine']
np.testing.assert_array_equal(
X_copy.data[0],
np.array([1, 2, 3], dtype=np.float32)
)
assert X_copy.attrs['a'] == [1, 2]
assert X_copy.attrs['b'] == -1
X.name = 'Elon 2'
X.type[0] = 'Robot'
X.data[0] *= 2
X.attrs['a'].append(3)
assert X.name == 'Elon 2'
assert X.type == ['Robot']
np.testing.assert_array_equal(
X.data[0],
np.array([2, 4, 6], dtype=np.float32)
)
assert X.attrs['a'] == [1, 2, 3]
assert X.attrs['b'] == -1
assert X_copy.name == 'Elon'
assert X_copy.type == ['Machine']
np.testing.assert_array_equal(
X_copy.data[0],
np.array([1, 2, 3], dtype=np.float32)
)
assert X_copy.attrs['a'] == [1, 2]
assert X_copy.attrs['b'] == -1
def test_copy(self):
X = XLayer(
name="Elon",
type=["Machine"],
data=[np.array([1, 2, 3], dtype=np.float32)],
attrs={
"a": [1, 2],
"b": -1
},
)
X_copy = X.copy()
assert X_copy.name == "Elon"
assert X_copy.type == ["Machine"]
np.testing.assert_array_equal(X_copy.data[0],
np.array([1, 2, 3], dtype=np.float32))
assert X_copy.attrs["a"] == [1, 2]
assert X_copy.attrs["b"] == -1
X.name = "Elon 2"
X.type[0] = "Robot"
X.data[0] *= 2
X.attrs["a"].append(3)
assert X.name == "Elon-2"
assert X.type == ["Robot"]
np.testing.assert_array_equal(X.data[0],
np.array([2, 4, 6], dtype=np.float32))
assert X.attrs["a"] == [1, 2, 3]
assert X.attrs["b"] == -1
assert X_copy.name == "Elon"
assert X_copy.type == ["Machine"]
np.testing.assert_array_equal(X_copy.data[0],
np.array([1, 2, 3], dtype=np.float32))
assert X_copy.attrs["a"] == [1, 2]
assert X_copy.attrs["b"] == -1
