Example #1
0
 def save(self, filename, compression=True, transpose=False,
          sparse=False, support=False, compression_opts=1):
     """
     Saves the array to a ga file
     """
     write_ga_file(filename, self.value, self[0].layout.metric, self[0].layout.basis_names,
                   compression=compression, transpose=transpose,
                   sparse=sparse, support=support, compression_opts=compression_opts)
Example #2
0
    def test_write_and_read(self, tmp_path, rng):  # noqa: F811
        file_name = str(tmp_path / "test.ga")

        basis_names = np.array(layout.basis_names, dtype=str)

        mv_array = ConformalMVArray([random_point_pair(rng=rng) 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)