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