Esempio n. 1
0
    def testSlicerEquivalence(self):
        """Test that slicers are marked equal when appropriate, and unequal when appropriate."""
        # Note that opsimfield slicers are considered 'equal' when all fieldId's, RA and Decs match.
        testslicer2 = OpsimFieldSlicer()
        fieldData2 = np.copy(self.fieldData)
        testslicer2.setupSlicer(self.simData, fieldData2)
        # These slicers should be equal.
        self.assertTrue(self.testslicer == testslicer2)
        self.assertFalse(self.testslicer != testslicer2)
        # These slicers should not be equal.
        fieldData2['fieldId'] = fieldData2['fieldId'] + 1
        testslicer2.setupSlicer(self.simData, fieldData2)
        self.assertTrue(self.testslicer != testslicer2)
        self.assertFalse(self.testslicer == testslicer2)
        # Test a slicer that is not the same kind.
        testslicer2 = UniSlicer()
        self.assertNotEqual(self.testslicer, testslicer2)
        # Test slicers that haven't been setup
        ts1 = OpsimFieldSlicer()
        ts2 = OpsimFieldSlicer()

        self.assertTrue(ts1 == ts2)
        self.assertFalse(ts1 != ts2)
        # Set up one with an odd value.
        ts2 = OpsimFieldSlicer(fieldRaColName='WackyName')
        self.assertTrue(ts1 != ts2)
        self.assertFalse(ts1 == ts2)
    def setUp(self):
        self.testslicer = OpsimFieldSlicer()
        self.fieldData = makeFieldData()
        self.simData = makeDataValues(self.fieldData)
        self.testslicer.setupSlicer(self.simData, self.fieldData)

        self.metricdata = ma.MaskedArray(data = np.zeros(len(self.testslicer), dtype='float'),
                                         mask = np.zeros(len(self.testslicer), 'bool'),
                                         fill_value = self.testslicer.badval)
        for i, s in enumerate(self.testslicer):
            idxs = s['idxs']
            if len(idxs) > 0:
                self.metricdata.data[i] = np.mean(self.simData['testdata'][idxs])
            else:
                self.metricdata.mask[i] = True
        self.metricdata2 = ma.MaskedArray(data = np.random.rand(len(self.testslicer)),
                                          mask = np.zeros(len(self.testslicer), 'bool'),
                                          fill_value = self.testslicer.badval)
Esempio n. 3
0
 def setUp(self):
     self.testslicer = OpsimFieldSlicer()
     self.fieldData = makeFieldData()
     self.simData = makeDataValues(self.fieldData, random=98)
Esempio n. 4
0
 def setUp(self):
     self.testslicer = OpsimFieldSlicer(latLonDeg=True)
     self.fieldData = makeFieldData()
     self.simData = makeDataValues(self.fieldData, random=776221)
     self.testslicer.setupSlicer(self.simData, self.fieldData)
Esempio n. 5
0
 def setUp(self):
     self.testslicer = OpsimFieldSlicer()
     self.fieldData = makeFieldData()
     self.simData = makeDataValues(self.fieldData)
     self.testslicer.setupSlicer(self.simData, self.fieldData)