#!/bin/bash
set -e

out=${1:-output2}

echo "" > $out

echo "Below is the feedback and provisional mark for your submission" >> $out
echo "for assignment 6 (Part 3).  Please note all marks are provisional until" >> $out
echo "ratified by the assessment board -- this is not an official" >> $out
echo "results transcript." >> $out
echo "" >> $out


#alarm() { perl -e 'alarm shift; exec @ARGV' "$@"; }
#alarm 20 foo arg1

# compilation tests

function scala_compile {
  (scala "$1" 2> /dev/null 1> /dev/null) 
}



# functional tests

function scala_assert {
  (scala -i "$1" "$2" -e "" 2> /dev/null 1> /dev/null)
}


# marks for CW1 part 3
marks=$(( 0 ))


# compilation test
echo "drumb.scala runs?" | tee -a $out

if (scala_compile drumb.scala)
then
  echo "  --> yes" | tee -a $out
  tsts=$(( 0 ))
else
  echo "  --> scala did not run drumb.scala" | tee -a $out
  tsts=$(( 1 )) 
fi


### get price tests

if [ $tsts -eq 0 ]
then
  echo "  get_prices(List(\"BIDU\"), 2004 to 2008) =" | tee -a $out
  echo "  List(List(None), List(None), List(Some(6.35))," | tee -a $out
  echo "       List(Some(12.241)), List(Some(38.188)))" | tee -a $out

  if (scala_assert "drumb.scala" "../../../marking/drumb_test1.scala")
  then
    echo "  --> success" | tee -a $out
    marks=$(( marks + 1 ))
  else
    echo "  --> test failed" | tee -a $out
  fi
fi

### get_deltas_test

if [ $tsts -eq 0 ]
then
  echo "  get_deltas(get_prices(List(\"IBM\", \"BIDU\"), 2004 to 2008)) = " | tee -a $out
  echo "  List(List(Some(0.07596697626574789), None), " | tee -a $out
  echo "       List(Some(-0.152620823795232), None)," | tee -a $out
  echo "       List(Some(0.20126676681483952), Some(0.9277165354330709))," | tee -a $out
  echo "       List(Some(0.09141762603007778), Some(2.119679764725104)))" | tee -a $out
  
  if (scala_assert "drumb.scala" "../../../marking/drumb_test2.scala") 
  then
    echo "  --> success" | tee -a $out
    marks=$(( marks + 1 ))
  else
    echo "  --> test failed" | tee -a $out
  fi
fi

### yield_tests, investmetn_test

if [ $tsts -eq 0 ]
then
  echo "  yearly_yield(IBM + BIDU, 04 - 08, 100, 0) - 107).abs <= 2 " | tee -a $out
  echo "  yearly_yield(IBM + BIDU, 04 - 08, 100, 1) -  85).abs <= 2 " | tee -a $out
  echo "  yearly_yield(IBM + BIDU, 04 - 08, 100, 2) - 156).abs <= 2 " | tee -a $out
  echo "  yearly_yield(IBM + BIDU, 04 - 08, 100, 3) - 210).abs <= 2 " | tee -a $out
  echo "  investment(List(\"IBM\", \"BIDU\"), 2004 to 2008, 100) - 298).abs <= 10" | tee -a $out
  
  if (scala_assert "drumb.scala" "../../../marking/drumb_test3.scala") 
  then
    echo "  --> success" | tee -a $out
    marks=$(( marks + 1 ))
  else
    echo "  --> test failed" | tee -a $out
  fi
fi

## final marks
echo "Overall mark for CW 1, Part 3:" | tee -a $out
echo "$marks" | tee -a $out