In [2]:
import unittest
import sys

class MyTest(unittest.TestCase):
    pass

In [3]:
%run ../../utils/load_notebook.py # импортируем функцию загрузки ноутбука

In [4]:
from photometry import * # загружаем функции


importing Jupyter notebook from photometry.ipynb
Using matplotlib backend: Qt4Agg
Populating the interactive namespace from numpy and matplotlib

In [5]:
#TODO: добавить тест на калибровку
bell_mass_to_light(1., 'I', 'B-R')


Out[5]:
1.297179270983956

In [6]:
MyTest.test1 = lambda self: self.assertAlmostEqual(bell_mass_to_light(1., 'I', 'B-R'), 1.297179270983)
TEST:
Тест на то, что обе функции поверхностных яркостей для балджа дают одинаковые значения до четвертого знака:

In [10]:
mu_eff = 18.37
r_eff = 8.8
n = 2.3

MyTest.test2 = lambda self: self.assertAlmostEqual(mu_bulge(10., mu_eff=mu_eff, r_eff=r_eff, n=n), 
                                                  mu_bulge2(10., mu_eff=mu_eff, r_eff=r_eff, n=n), places=3)

Запуск всех тестов:


In [11]:
suite = unittest.TestLoader().loadTestsFromTestCase(MyTest)
unittest.TextTestRunner(verbosity=1, stream=sys.stderr).run(suite)


..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK
Out[11]:
<unittest.runner.TextTestResult run=2 errors=0 failures=0>