casa6

import unittest

import os

import json

import shutil

​

from casatools import ctsys 

from casatestutils.stakeholder import almastktestutils

from casatestutils import compare

​

class Test_almastkutils(unittest.TestCase):

    @classmethod

    def setUpClass(cls):

        ''' common input data '''

        cls.datapath = ctsys.resolve('stakeholder/alma')

        cls.oldtestscr = 'test_stk_alma_pipeline_imaging_old.py'

        cls.curjson = 'test_stk_alma_pipeline_imaging_exp_dicts.json'

        cls.stdcubejson = 'test_standard_cube_cur_stats.json'

        cls.updatedjson = 'update_exp_subdicts.json'

    def setUp(self):

        shutil.copy(os.path.join(self.datapath,self.stdcubejson),self.stdcubejson)

        shutil.copy(os.path.join(self.datapath,self.curjson),self.curjson)

        shutil.copy(os.path.join(self.datapath,self.updatedjson), self.updatedjson)

​

    def compareDict(self, indict, refdict):

        ''' compare metric dictionaries, allow the comparison on vlaues between exp_dicts with extra flags and current metric dicts '''

        nfailcontent=0

        for key in indict:

            if key in refdict:

                if isinstance(indict[key],dict):

                    # additional dict level

                    for mt in indict[key]:

                        if mt in refdict[key]:

                            if isinstance(refdict[key][mt],list) and not isinstance(indict[key][mt],list):

                                if indict[key][mt] == refdict[key][mt][1]:

                                    pass

                                else:

                                    nfailcontent += 1

                            else:

                                if indict[key][mt] == refdict[key][mt]:

                                    pass

                                else:

                                    nfailcontent += 1

                        else:

                            nfailcontent += 1

                else: 

                   if isinstance(refdict[key],list) and not isinstance(indict[key],list):

                      if indict[key] == refdict[key][1]:

                          pass

                      else:

                          nfailcontent += 1

                   else:

                      if indict[key]==refdict[key]:

                          pass 

                      else:

                           nfailcontent += 1

        return nfailcontent

    def test_extract_subexpdict(self):

        ''' Test a fuction to extract a subset of exp dicts from a single testcase json file '''

        print("stdcubejon=",self.stdcubejson)

        metrickeys=['im_stats_dict','psf_stats_dict']