Source code for pyretis.testing.systemhelp

# -*- coding: utf-8 -*-
# Copyright (c) 2021, PyRETIS Development Team.
# Distributed under the LGPLv2.1+ License. See LICENSE for more info.
"""Methods that might be useful for testing.

This module defines methods that are useful in connection with

from pyretis.core.system import System
from pyretis.core.particles import ParticlesExt

[docs]def create_system_ext(pos=None, vel=False): """Create an external system with given positions and velocities.""" system = System() system.particles = ParticlesExt(dim=3) if pos is None: system.particles.set_pos((None, None)) else: system.particles.set_pos(pos) system.particles.set_vel(vel) return system