#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import print_function, division

import unittest as ut
import numpy as np

import ARG.arg as arg


class ARGTestCase(ut.TestCase):

    def test_truisms(self):
        """Test parameter class"""
        param = arg.ARGparams()
        self.assertTrue(isinstance(param.scale, float))
        self.assertTrue(isinstance(param.rho, float))
        self.assertTrue(isinstance(param.delta, float))
        self.assertTrue(isinstance(param.beta(), float))


if __name__ == '__main__':
    ut.main()
@@ -204,14 +204,18 @@
 est_
-truism
+param_clas
 s(se
@@ -254,12 +254,14 @@
 lass
+.
 %22%22%22%0A
+%0A
     
@@ -303,32 +303,27 @@
  self.assert
-True(isi
+IsI
 nstance(para
@@ -329,33 +329,32 @@
 am.scale, float)
-)
 %0A        self.as
@@ -353,32 +353,27 @@
  self.assert
-True(isi
+IsI
 nstance(para
@@ -377,33 +377,32 @@
 aram.rho, float)
-)
 %0A        self.as
@@ -401,32 +401,27 @@
  self.assert
-True(isi
+IsI
 nstance(para
@@ -435,17 +435,16 @@
 , float)
-)
 %0A       
@@ -459,44 +459,385 @@
 sert
-True(isinstance(param.beta(), float)
+IsInstance(param.beta(), float)%0A%0A    def test_abc_functions(self):%0A        %22%22%22Test functions a, b, c of ARG model.%22%22%22%0A%0A        argmodel = arg.ARG()%0A        uarg = np.linspace(-50, 100, 100)%0A        self.assertIsInstance(argmodel.afun(uarg), np.ndarray)%0A        self.assertIsInstance(argmodel.bfun(uarg), np.ndarray)%0A        self.assertIsInstance(argmodel.cfun(uarg), np.ndarray
 )%0A%0A%0A

Test functions a, b, c