quantified.uncertainty/squiggle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

it string in tests

Browse Source
This commit is contained in:
Quinn Dougherty 2022-09-01 12:54:04 +08:00
parent 9366ce61f3
commit ab5b54413b
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/squiggle-lang/__tests__/Stdlib_test.res
View File

@ -7,10 +7,14 @@ let makeTest = (~only=false, str, item1, item2) =>
: test(str, () => expect(item1)->toEqual(item2)) : test(str, () => expect(item1)->toEqual(item2))
describe("Stdlib", () => { describe("Stdlib", () => {
makeTest("min", Stdlib.Random.sample([1.0, 2.0], {probs: [0.5, 0.5], size: 10}) |> E.A.length, 10)
makeTest( makeTest(
"min", "Length of Random.sample",
Stdlib.Random.sample([1.0, 2.0], {probs: [0.5, 0.5], size: 10}) |> E.A.uniq |> E.A.Floats.sort, Stdlib.Random.sample([1.0, 2.0], {probs: [0.5, 0.5], size: 10})->E.A.length,
10,
)
makeTest(
"Random.sample returns elements from input array (will fail with very slim probability)",
Stdlib.Random.sample([1.0, 2.0], {probs: [0.5, 0.5], size: 10})->E.A.uniq->E.A.Floats.sort,
[1.0, 2.0], [1.0, 2.0],
) )
}) })

2
packages/squiggle-lang/src/rescript/Distributions/PointSetDist/Discrete.res
View File

@ -226,6 +226,6 @@ module T = Dist({
}) })
let sampleN = (t: t, n): array<float> => { let sampleN = (t: t, n): array<float> => {
let normalized = t |> T.normalize |> getShape let normalized = t->T.normalize->getShape
Stdlib.Random.sample(normalized.xs, {probs: normalized.ys, size: n}) Stdlib.Random.sample(normalized.xs, {probs: normalized.ys, size: n})
} }