aboutsummaryrefslogtreecommitdiff
path: root/stdlib/source/test/lux/math/modulus.lux
blob: c5147e75cf0c14a0241ff73bf1ff5813e127661c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
(.module:
  [lux #*
   ["_" test (#+ Test)]
   ["." meta]
   [abstract
    [monad (#+ do)]]
   [control
    ["." try]
    ["." exception]]
   [math
    ["." random (#+ Random)]
    [number
     ["i" int]]]
   [macro
    [syntax (#+ syntax:)]
    ["." code]]]
  {1
   ["." /]})

(syntax: (|divisor|)
  (do meta.monad
    [divisor meta.count]
    (wrap (list (code.int (case divisor
                            0 +1
                            _ (.int divisor)))))))

(def: #export (random range)
  (Ex [%] (-> Int (Random (/.Modulus %))))
  (|> random.int
      (\ random.monad map (i.% range))
      (random.one (|>> /.modulus try.to_maybe))))

(def: #export test
  Test
  (<| (_.covering /._)
      (_.for [/.Modulus])
      (do random.monad
        [divisor random.int
         modulus (..random +1,000,000)
         dividend random.int]
        ($_ _.and
            (_.cover [/.modulus /.divisor]
                     (case (/.modulus divisor)
                       (#try.Success modulus)
                       (i.= divisor (/.divisor modulus))
                       
                       (#try.Failure error)
                       (i.= +0 divisor)))
            (_.cover [/.zero_cannot_be_a_modulus]
                     (case (/.modulus +0)
                       (#try.Failure error)
                       (exception.match? /.zero_cannot_be_a_modulus error)

                       (#try.Success modulus)
                       false))
            (_.cover [/.literal]
                     (with_expansions [<divisor> (|divisor|)]
                       (i.= <divisor> (/.divisor (/.literal <divisor>)))))
            (_.cover [/.=]
                     (with_expansions [<divisor> (|divisor|)]
                       (/.= (/.literal <divisor>)
                            (/.literal <divisor>))))
            (_.cover [/.congruent?]
                     (and (/.congruent? modulus dividend dividend)
                          (or (not (/.congruent? modulus dividend (inc dividend)))
                              (i.= +1 (/.divisor modulus)))))
            ))))