diff --git a/unittests/testBaseline.py b/unittests/testBaseline.py new file mode 100644 index 0000000..9926123 --- /dev/null +++ b/unittests/testBaseline.py @@ -0,0 +1,27 @@ + +import unittest +import numpy as np +import helperFunctions as hF +import matplotlib.pyplot as plt +from CellData import icelldata_of_dir +from Baseline import BaselineCellData +import os + + +class BaselineTester(unittest.TestCase): + + def setUp(self): + pass + + def tearDown(self): + pass + + def test_get_burstiness(self): + for cell in icelldata_of_dir("../data/"): + + baseline = BaselineCellData(cell) + + bursty = baseline.get_burstiness() + + self.assertTrue(bursty <= 1) + self.assertTrue(bursty >= 0)