pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Review actions for monte carlo models.

Browse source
This commit is contained in:
Phil Elson 2021-05-28 17:23:43 +02:00
parent 9669e5afd0
commit 38fe6e734e
2 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cara/monte_carlo/sampleable.py
View file

@ -16,12 +16,12 @@ class SampleableDistribution:
class Normal(SampleableDistribution):
def __init__(self, mean: float, scale: float):
def __init__(self, mean: float, standard_deviation: float):
self.mean = mean
self.scale = scale
self.standard_deviation = standard_deviation
def generate_samples(self, size: int) -> float_array_size_n:
return np.random.normal(self.mean, self.scale, size=size)
return np.random.normal(self.mean, self.standard_deviation, size=size)
_VectorisedFloatOrSampleable = typing.Union[

4
cara/tests/test_monte_carlo.py
View file

@ -1,5 +1,6 @@
import dataclasses
import numpy as np
import pytest
import cara.models
@ -81,3 +82,6 @@ def test_build_concentration_model(baseline_mc_model: cara.monte_carlo.Concentra
def test_build_exposure_model(baseline_mc_exposure_model: cara.monte_carlo.ExposureModel):
model = baseline_mc_exposure_model.build_model(7)
assert isinstance(model, cara.models.ExposureModel)
prob = model.quanta_exposure()
assert isinstance(prob, np.ndarray)
assert prob.shape == (7, )