Skip to content

Commit

Permalink
poly: fix eval and sorted_3_ + fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
PottierLoic committed Jul 15, 2024
1 parent 7fd1d43 commit 4c41c75
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 23 deletions.
18 changes: 10 additions & 8 deletions poly/poly.v
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,11 @@ pub fn eval(c []f64, x f64) f64 {
panic('coeficients can not be empty')
}
len := c.len
mut ans := c[len - 1]
for e in c[..len - 1] {
ans = e + x * ans
mut ans := 0.0
mut i := len-1
for i >= 0{
ans = c[i] + x * ans
i--
}
return ans
}
Expand Down Expand Up @@ -131,13 +133,13 @@ fn sorted_3_(x_ f64, y_ f64, z_ f64) (f64, f64, f64) {
mut y := y_
mut z := z_
if x > y {
y, x = swap_(x, y)
x, y = swap_(x, y)
}
if y > z {
z, y = swap_(y, z)
if x > z {
x, z = swap_(x, z)
}
if x > y {
y, x = swap_(x, y)
if y > z {
y, z = swap_(y, z)
}
return x, y, z
}
Expand Down
31 changes: 16 additions & 15 deletions poly/poly_test.v
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@ module poly

fn test_eval() {
// ans = 2
// ans = 4.0 + 4 * 2 = 12
// ans = 5 + 4 * 12 = 53
// ans = 5 + 4 * 2 = 13
// ans = 4 + 4 * 13 = 56
x := 4
cof := [4.0, 5, 2]
assert eval(cof, 4) == 53
assert eval(cof, 4) == 56
}

fn test_swap() {
Expand All @@ -16,13 +16,13 @@ fn test_swap() {
assert a == 202.0 && b == 101.0
}

// fn test_sorted_3_(){
// a := 5.0
// b := 7.0
// c := -8.0
// x, y, z := sorted_3_(a, b, c)
// assert y == 7.0
// }
fn test_sorted_3_(){
a := 5.0
b := 7.0
c := -8.0
x, y, z := sorted_3_(a, b, c)
assert y == 5.0
}

fn test_add() {
a := [6.0, 777, -3]
Expand All @@ -36,9 +36,10 @@ fn test_substract() {
assert substract(a, b) == [5.0, 1532, 1]
}

// fn test_multiply(){
// a := [9.0, -1, 5]
// b := [0.0, -1, 7]
fn test_multiply(){
a := [9.0, -1, 5]
b := [0.0, -1, 7]

assert multiply(a, b) == [0.0, -9, 64, -12, 35, 0]
}

// assert multiply(a, b) == [0.0, -9, 73, -12, 35, 0]
// }

0 comments on commit 4c41c75

Please sign in to comment.