示例#1
0
    def test_create(self):
        obs_data = ObsData()
        obs_size = 10
        block = obs_data.addBlock("OBS", obs_size)
        self.assertTrue(isinstance(block, ObsBlock))

        block[0] = (10, 10)
        block[1] = (12, 12)
        D = obs_data.createDObs()
        self.assertTrue(isinstance(D, Matrix))
        self.assertEqual(D.dims(), (2, 1))

        self.assertEqual(D[0, 0], 10)
        self.assertEqual(D[1, 0], 12)

        obs_data.scaleMatrix(D)
        self.assertEqual(D[0, 0], 1)
        self.assertEqual(D[1, 0], 1)

        R = obs_data.createR()
        self.assertEqual((2, 2), R.dims())

        with self.assertRaises(IndexError):
            obs_data[10]

        v, s = obs_data[0]
        self.assertEqual(v, 10)
        self.assertEqual(s, 10)

        v, s = obs_data[1]
        self.assertEqual(v, 12)
        self.assertEqual(s, 12)
示例#2
0
    def test_create(self):
        obs_data = ObsData()
        obs_size = 10
        block = obs_data.addBlock("OBS" , obs_size)
        self.assertTrue( isinstance( block , ObsBlock ))

        block[0] = (10,10)
        block[1] = (12,12)
        D = obs_data.createDObs()
        self.assertTrue( isinstance(D , Matrix ))
        self.assertEqual( D.dims() , (2,1))
        
        self.assertEqual( D[0,0] , 10 )
        self.assertEqual( D[1,0] , 12 )
        
        obs_data.scaleMatrix( D )
        self.assertEqual( D[0,0] , 1 )
        self.assertEqual( D[1,0] , 1 )
        
        R = obs_data.createR()
        self.assertEqual( (2,2) , R.dims() )

        with self.assertRaises(IndexError):
            obs_data[10]

        v,s = obs_data[0]
        self.assertEqual( v , 10 )
        self.assertEqual( s , 10 )


        v,s = obs_data[1]
        self.assertEqual( v , 12 )
        self.assertEqual( s , 12 )