How to use plonk - 10 common examples

def test_read_header(self):
        """Testing reading Phantom HDF5 dump header."""

        dump = plonk.Dump(TEST_FILE)

        for para in dump.header:
            if isinstance(dump.header[para], np.ndarray):
                np.testing.assert_allclose(dump.header[para], header[para])
            else:
                self.assertEqual(dump.header[para], header[para])

        dump._close_file()

def test_read_dump(self):
        """Testing reading Phantom HDF5 dumps."""

        # Read from pathlib.Path
        dump = plonk.Dump(TEST_FILE)
        dump._close_file()

        # Read from str
        dump = plonk.Dump(str(TEST_FILE))
        dump._close_file()

        # Not exists
        self.assertRaises(FileNotFoundError, plonk.Dump, 'does_not_exist.h5')
        dump._close_file()

def test_load_arrays_into_memory(self):

        dump = plonk.Dump(TEST_FILE)
        dump._load_arrays('sinks')

        dump._close_file()

def test_read_dump(self):
        """Testing reading Phantom HDF5 dumps."""

        # Read from pathlib.Path
        dump = plonk.Dump(TEST_FILE)
        dump._close_file()

        # Read from str
        dump = plonk.Dump(str(TEST_FILE))
        dump._close_file()

        # Not exists
        self.assertRaises(FileNotFoundError, plonk.Dump, 'does_not_exist.h5')
        dump._close_file()

def test_initialization(self):

        dump = plonk.Dump(TEST_FILE)
        options = list()
        options.append({'render': 'density'})
        options.append({'render': 'divv'})

        plonk.visualization.MultiPlot(dump, options)

        dumps = [dump, dump]
        options = {'render': 'density', 'extent': [-100, 100, -100, 100]}

        plonk.visualization.MultiPlot(dumps, options)

def test_extra_quantity(self):

        dump = plonk.Dump(TEST_FILE)

        self.assertAlmostEqual(
            dump.extra_quantity('L', sph_type='particles')[0].mean(),
            6.875907667109459e-05,
        )

        self.assertAlmostEqual(
            dump.extra_quantity('p')[0].mean(), -3.029025112420847e-22
        )

        self.assertAlmostEqual(
            dump.extra_quantity('R', sph_type='sinks')[0].mean(), 3.180317954809203e-15
        )

        dump._close_file()

def test_density(self):

        dump = plonk.Dump(TEST_FILE)

        self.assertAlmostEqual(dump.density.mean(), 4.739748038277296e-08)

        dump._close_file()

def test_read_particle_arrays(self):
        """Testing reading Phantom HDF5 dump particle arrays."""

        dump = plonk.Dump(TEST_FILE)

        self.assertEqual(set(dump.particles.arrays), array_keys)

        for key, val in dump.particles.arrays.items():
            np.testing.assert_allclose(val[()].mean(), mean_array_values[key])

        dump._close_file()

def test_load_phantom_snap():
    """Testing reading Phantom HDF5 snapshots."""
    # Read from pathlib.Path
    snap = plonk.load_snap(TEST_FILE)
    snap.close_file()
    # Read from str
    snap = plonk.load_snap(str(TEST_FILE))
    snap.close_file()
    # Not exists
    with pytest.raises(FileNotFoundError):
        plonk.load_snap('does_not_exist.h5')

def test_initialization():
    """Test plot function."""
    snap = plonk.load_snap(TEST_FILE)
    plonk.visualize.plot(
        snap=snap,
        quantity='density',
        x='x',
        y='y',
        extent=(-150, 150, -150, 150),
        norm='linear',
        cmap='gist_heat',
        number_of_pixels=(512, 512),
    )