extra/math/blas/vectors/vectors-tests.factor

   1 USING: kernel math.blas.vectors math.blas.syntax sequences tools.test ;
   2 IN: math.blas.vectors.tests
   3
   4 ! clone
   5
   6 [ svector{ 1.0 2.0 3.0 } ] [ svector{ 1.0 2.0 3.0 } clone ] unit-test
   7 [ f ] [ svector{ 1.0 2.0 3.0 } dup clone eq? ] unit-test
   8 [ dvector{ 1.0 2.0 3.0 } ] [ dvector{ 1.0 2.0 3.0 } clone ] unit-test
   9 [ f ] [ dvector{ 1.0 2.0 3.0 } dup clone eq? ] unit-test
  10 [ cvector{ 1.0 C{ 2.0 3.0 } 4.0 } ] [ cvector{ 1.0 C{ 2.0 3.0 } 4.0 } clone ] unit-test
  11 [ f ] [ cvector{ 1.0 C{ 2.0 3.0 } 4.0 } dup clone eq? ] unit-test
  12 [ zvector{ 1.0 C{ 2.0 3.0 } 4.0 } ] [ zvector{ 1.0 C{ 2.0 3.0 } 4.0 } clone ] unit-test
  13 [ f ] [ zvector{ 1.0 C{ 2.0 3.0 } 4.0 } dup clone eq? ] unit-test
  14
  15 ! nth
  16
  17 [ 1.0 ] [ 2 svector{ 3.0 2.0 1.0 } nth ] unit-test
  18 [ 1.0 ] [ 2 dvector{ 3.0 2.0 1.0 } nth ] unit-test
  19
  20 [ C{ 1.0 2.0 } ]
  21 [ 2 cvector{ C{ -3.0 -2.0 } C{ -1.0 0.0 } C{ 1.0 2.0 } } nth ] unit-test
  22
  23 [ C{ 1.0 2.0 } ]
  24 [ 2 zvector{ C{ -3.0 -2.0 } C{ -1.0 0.0 } C{ 1.0 2.0 } } nth ] unit-test
  25
  26 ! set-nth
  27
  28 [ svector{ 3.0 2.0 0.0 } ] [ 0.0 2 svector{ 3.0 2.0 1.0 } [ set-nth ] keep ] unit-test
  29 [ dvector{ 3.0 2.0 0.0 } ] [ 0.0 2 dvector{ 3.0 2.0 1.0 } [ set-nth ] keep ] unit-test
  30
  31 [ cvector{ C{ -3.0 -2.0 } C{ -1.0 0.0 } C{ 3.0 4.0 } } ] [
  32     C{ 3.0 4.0 } 2
  33     cvector{ C{ -3.0 -2.0 } C{ -1.0 0.0 } C{ 1.0 2.0 } }
  34     [ set-nth ] keep
  35 ] unit-test
  36 [ zvector{ C{ -3.0 -2.0 } C{ -1.0 0.0 } C{ 3.0 4.0 } } ] [
  37     C{ 3.0 4.0 } 2
  38     zvector{ C{ -3.0 -2.0 } C{ -1.0 0.0 } C{ 1.0 2.0 } }
  39     [ set-nth ] keep
  40 ] unit-test
  41
  42 ! V+
  43
  44 [ svector{ 11.0 22.0 } ] [ svector{ 1.0 2.0 } svector{ 10.0 20.0 } V+ ] unit-test
  45 [ dvector{ 11.0 22.0 } ] [ dvector{ 1.0 2.0 } dvector{ 10.0 20.0 } V+ ] unit-test
  46
  47 [ cvector{ 11.0 C{ 22.0 33.0 } } ]
  48 [ cvector{ 1.0 C{ 2.0 3.0 } } cvector{ 10.0 C{ 20.0 30.0 } } V+ ]
  49 unit-test
  50
  51 [ zvector{ 11.0 C{ 22.0 33.0 } } ]
  52 [ zvector{ 1.0 C{ 2.0 3.0 } } zvector{ 10.0 C{ 20.0 30.0 } } V+ ]
  53 unit-test
  54
  55 ! V-
  56
  57 [ svector{ 9.0 18.0 } ] [ svector{ 10.0 20.0 } svector{ 1.0 2.0 } V- ] unit-test
  58 [ dvector{ 9.0 18.0 } ] [ dvector{ 10.0 20.0 } dvector{ 1.0 2.0 } V- ] unit-test
  59
  60 [ cvector{ 9.0 C{ 18.0 27.0 } } ]
  61 [ cvector{ 10.0 C{ 20.0 30.0 } } cvector{ 1.0 C{ 2.0 3.0 } } V- ]
  62 unit-test
  63
  64 [ zvector{ 9.0 C{ 18.0 27.0 } } ]
  65 [ zvector{ 10.0 C{ 20.0 30.0 } } zvector{ 1.0 C{ 2.0 3.0 } } V- ]
  66 unit-test
  67
  68 ! Vneg
  69
  70 [ svector{ 1.0 -2.0 } ] [ svector{ -1.0 2.0 } Vneg ] unit-test
  71 [ dvector{ 1.0 -2.0 } ] [ dvector{ -1.0 2.0 } Vneg ] unit-test
  72
  73 [ cvector{ 1.0 C{ -2.0 3.0 } } ] [ cvector{ -1.0 C{ 2.0 -3.0 } } Vneg ] unit-test
  74 [ zvector{ 1.0 C{ -2.0 3.0 } } ] [ zvector{ -1.0 C{ 2.0 -3.0 } } Vneg ] unit-test
  75
  76 ! n*V
  77
  78 [ svector{ 100.0 200.0 } ] [ 10.0 svector{ 10.0 20.0 } n*V ] unit-test
  79 [ dvector{ 100.0 200.0 } ] [ 10.0 dvector{ 10.0 20.0 } n*V ] unit-test
  80
  81 [ cvector{ C{ 20.0 4.0 } C{ 8.0 12.0 } } ]
  82 [ C{ 10.0 2.0 } cvector{ 2.0 C{ 1.0 1.0 } } n*V ]
  83 unit-test
  84
  85 [ zvector{ C{ 20.0 4.0 } C{ 8.0 12.0 } } ]
  86 [ C{ 10.0 2.0 } zvector{ 2.0 C{ 1.0 1.0 } } n*V ]
  87 unit-test
  88
  89 ! V*n
  90
  91 [ svector{ 100.0 200.0 } ] [ svector{ 10.0 20.0 } 10.0 V*n ] unit-test
  92 [ dvector{ 100.0 200.0 } ] [ dvector{ 10.0 20.0 } 10.0 V*n ] unit-test
  93
  94 [ cvector{ C{ 20.0 4.0 } C{ 8.0 12.0 } } ]
  95 [ cvector{ 2.0 C{ 1.0 1.0 } } C{ 10.0 2.0 } V*n ]
  96 unit-test
  97
  98 [ zvector{ C{ 20.0 4.0 } C{ 8.0 12.0 } } ]
  99 [ zvector{ 2.0 C{ 1.0 1.0 } } C{ 10.0 2.0 } V*n ]
 100 unit-test
 101
 102 ! V/n
 103
 104 [ svector{ 1.0 2.0 } ] [ svector{ 4.0 8.0 } 4.0 V/n ] unit-test
 105 [ dvector{ 1.0 2.0 } ] [ dvector{ 4.0 8.0 } 4.0 V/n ] unit-test
 106
 107 [ cvector{ C{ 0.0 -4.0 } 1.0 } ]
 108 [ cvector{ C{ 4.0 -4.0 } C{ 1.0 1.0 } } C{ 1.0 1.0 } V/n ]
 109 unit-test
 110
 111 [ zvector{ C{ 0.0 -4.0 } 1.0 } ]
 112 [ zvector{ C{ 4.0 -4.0 } C{ 1.0 1.0 } } C{ 1.0 1.0 } V/n ]
 113 unit-test
 114
 115 ! V.
 116
 117 [ 7.0 ] [ svector{ 1.0 2.5 } svector{ 2.0 2.0 } V. ] unit-test
 118 [ 7.0 ] [ dvector{ 1.0 2.5 } dvector{ 2.0 2.0 } V. ] unit-test
 119 [ C{ 7.0 7.0 } ] [ cvector{ C{ 1.0 1.0 } 2.5 } cvector{ 2.0 C{ 2.0 2.0 } } V. ] unit-test
 120 [ C{ 7.0 7.0 } ] [ zvector{ C{ 1.0 1.0 } 2.5 } zvector{ 2.0 C{ 2.0 2.0 } } V. ] unit-test
 121
 122 ! V.conj
 123
 124 [ C{ 7.0 3.0 } ] [ cvector{ C{ 1.0 1.0 } 2.5 } cvector{ 2.0 C{ 2.0 2.0 } } V.conj ] unit-test
 125 [ C{ 7.0 3.0 } ] [ zvector{ C{ 1.0 1.0 } 2.5 } zvector{ 2.0 C{ 2.0 2.0 } } V.conj ] unit-test
 126
 127 ! Vnorm
 128
 129 [ 5.0 ] [ svector{ 3.0 4.0 } Vnorm ] unit-test
 130 [ 5.0 ] [ dvector{ 3.0 4.0 } Vnorm ] unit-test
 131
 132 [ 13.0 ] [ cvector{ C{ 3.0 4.0 } 12.0 } Vnorm ] unit-test
 133 [ 13.0 ] [ zvector{ C{ 3.0 4.0 } 12.0 } Vnorm ] unit-test
 134
 135 ! Vasum
 136
 137 [ 6.0 ] [ svector{ 1.0 2.0 -3.0 } Vasum ] unit-test
 138 [ 6.0 ] [ dvector{ 1.0 2.0 -3.0 } Vasum ] unit-test
 139
 140 [ 15.0 ] [ cvector{ 1.0 C{ -2.0 3.0 } C{ 4.0 -5.0 } } Vasum ] unit-test
 141 [ 15.0 ] [ zvector{ 1.0 C{ -2.0 3.0 } C{ 4.0 -5.0 } } Vasum ] unit-test
 142
 143 ! Vswap
 144
 145 [ svector{ 2.0 2.0 } svector{ 1.0 1.0 } ]
 146 [ svector{ 1.0 1.0 } svector{ 2.0 2.0 } Vswap ]
 147 unit-test
 148
 149 [ dvector{ 2.0 2.0 } dvector{ 1.0 1.0 } ]
 150 [ dvector{ 1.0 1.0 } dvector{ 2.0 2.0 } Vswap ]
 151 unit-test
 152
 153 [ cvector{ 2.0 C{ 2.0 2.0 } } cvector{ C{ 1.0 1.0 } 1.0 } ]
 154 [ cvector{ C{ 1.0 1.0 } 1.0 } cvector{ 2.0 C{ 2.0 2.0 } } Vswap ]
 155 unit-test
 156
 157 [ zvector{ 2.0 C{ 2.0 2.0 } } zvector{ C{ 1.0 1.0 } 1.0 } ]
 158 [ zvector{ C{ 1.0 1.0 } 1.0 } zvector{ 2.0 C{ 2.0 2.0 } } Vswap ]
 159 unit-test
 160
 161 ! Viamax
 162
 163 [ 3 ] [ svector{ 1.0 -5.0 4.0 -6.0 -1.0 } Viamax ] unit-test
 164 [ 3 ] [ dvector{ 1.0 -5.0 4.0 -6.0 -1.0 } Viamax ] unit-test
 165 [ 0 ] [ cvector{ C{ 2.0 -5.0 } 4.0 -6.0 -1.0 } Viamax ] unit-test
 166 [ 0 ] [ zvector{ C{ 2.0 -5.0 } 4.0 -6.0 -1.0 } Viamax ] unit-test
 167
 168 ! Vamax
 169
 170 [ -6.0 ] [ svector{ 1.0 -5.0 4.0 -6.0 -1.0 } Vamax ] unit-test
 171 [ -6.0 ] [ dvector{ 1.0 -5.0 4.0 -6.0 -1.0 } Vamax ] unit-test
 172 [ C{ 2.0 -5.0 } ] [ cvector{ C{ 2.0 -5.0 } 4.0 -6.0 -1.0 } Vamax ] unit-test
 173 [ C{ 2.0 -5.0 } ] [ zvector{ C{ 2.0 -5.0 } 4.0 -6.0 -1.0 } Vamax ] unit-test
 174
 175 ! Vsub
 176
 177 [ svector{ -5.0 4.0 -6.0 } ] [ svector{ 1.0 -5.0 4.0 -6.0 -1.0 } 1 3 Vsub ] unit-test
 178 [ dvector{ -5.0 4.0 -6.0 } ] [ dvector{ 1.0 -5.0 4.0 -6.0 -1.0 } 1 3 Vsub ] unit-test
 179 [ cvector{ -5.0 C{ 4.0 3.0 } -6.0 } ] [ cvector{ 1.0 -5.0 C{ 4.0 3.0 } -6.0 -1.0 } 1 3 Vsub ] unit-test
 180 [ zvector{ -5.0 C{ 4.0 3.0 } -6.0 } ] [ zvector{ 1.0 -5.0 C{ 4.0 3.0 } -6.0 -1.0 } 1 3 Vsub ] unit-test