def test_write_and_read_array(self, tmp_path, rng): # noqa: F811 file_name = str(tmp_path / "test.ga") mv_array = MVArray([random_point_pair(rng=rng) for i in range(1000)]) mv_array.save(file_name, compression=True, transpose=False, sparse=False, support=False) loaded_array = layout.load_ga_file(file_name) np.testing.assert_equal(loaded_array.value, mv_array.value)
def test_write_and_read(self, tmp_path, rng): # noqa: F811 file_name = str(tmp_path / "test.ga.json") basis_names = np.array(layout.basis_names, dtype=str) mv_array = ConformalMVArray([random_point_pair(rng=rng) for i in range(1000)]).value write_json_file(file_name, mv_array, layout.metric, basis_names, compression=True, transpose=False, sparse=False, support=False) data_array, metric_2, basis_names_2, support = read_json_file(file_name) np.testing.assert_equal(data_array, mv_array) np.testing.assert_equal(layout.metric, metric_2) np.testing.assert_equal(basis_names, basis_names_2)
def test_write_and_read(self, tmp_path): file_name = str(tmp_path / "test.ga") basis_names = np.array(list(sorted(layout.basis_vectors.keys())), dtype=bytes) mv_array = ConformalMVArray([random_point_pair() for i in range(1000)]).value write_ga_file(file_name, mv_array, layout.metric, basis_names, compression=True, transpose=False, sparse=False, support=False) data_array, metric_2, basis_names_2, support = read_ga_file(file_name) np.testing.assert_equal(data_array, mv_array) np.testing.assert_equal(layout.metric, metric_2) np.testing.assert_equal(basis_names, basis_names_2)