File: utils.py

package info (click to toggle)
python-ase 3.22.1-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 14,344 kB
  • sloc: python: 126,379; xml: 946; makefile: 111; javascript: 47
file content (13 lines) | stat: -rw-r--r-- 306 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
 
from numpy.random import RandomState


class RandomCalculator:
    """Fake Calculator class."""
    def __init__(self):
        self.rng = RandomState(42)

    def get_forces(self, atoms):
        return self.rng.rand(len(atoms), 3)

    def get_dipole_moment(self, atoms):
        return self.rng.rand(3)