Skip to content

Commit b9eaf44

Browse files
committed
feat: add example test for rws algorithm
1 parent ff204d8 commit b9eaf44

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed
Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,30 @@
1-
package main
1+
package rws
22

33
import (
44
"fmt"
5-
"github.com/Ja7ad/algo/rws"
65
"log"
76
)
87

9-
func main() {
8+
func ExampleNewWeightedSelector() {
109
weightedItems := map[int]string{
1110
3: "Apple",
1211
1: "Banana",
1312
6: "Cherry",
1413
}
1514

16-
selector, err := rws.NewWeightedSelector(weightedItems)
15+
selector, err := NewWeightedSelector(weightedItems)
1716
if err != nil {
1817
log.Fatal(err)
1918
}
2019

2120
selectedItem, _ := selector.Pick()
2221
fmt.Println("Selected:", selectedItem)
22+
}
2323

24+
func ExampleNewAutoWeightedSelector() {
2425
items := []string{"Dog", "Cat", "Fish"}
2526

26-
autoSelector, err := rws.NewAutoWeightedSelector(items)
27+
autoSelector, err := NewAutoWeightedSelector(items)
2728
if err != nil {
2829
log.Fatal(err)
2930
}

0 commit comments

Comments
 (0)