| import numpy as np | |
| from mla import backend, to_device, to_numpy | |
| def test_default_dtype_is_float64_on_numpy(): | |
| if backend.NAME == "numpy": | |
| assert backend.DTYPE == np.dtype("float64") | |
| def test_roundtrip_preserves_values(): | |
| a = np.array([[1.0, 2.0], [3.0, 4.0]]) | |
| b = to_numpy(to_device(a)) | |
| assert np.allclose(b, a) | |