def testLogExceptionsWithSystemExitExceptionAndDefaultLogger(self): loggerMock = Mock(spec_set=decorators.logging.getLogger()) with patch.object(decorators.logging, "getLogger", autospec=True, return_value=loggerMock): # SystemExit is based on BaseException, so we want to make sure that # those are handled properly, too inputArgs = (1, 2, 3) inputKwargs = dict(a="A", b="B", c="C") @decorators.logExceptions() def doSomething(*args, **kwargs): self.assertEqual(args, inputArgs) self.assertEqual(kwargs, inputKwargs) raise SystemExit() with self.assertRaises(SystemExit): doSomething(*inputArgs, **inputKwargs) self.assertEqual(loggerMock.exception.call_count, 1) self.assertIn("Unhandled exception %r from %r. Caller stack:\n%s", loggerMock.exception.call_args[0][0]) if __name__ == '__main__': unittest.main()
} state.__setstate__(record) self.assertEqual(state.ROWID, record['ROWID']) self.assertEqual(state.anomalyScore, record['anomalyScore']) self.assertEqual(state.anomalyVector, record['anomalyVector']) self.assertEqual(state.anomalyLabel, record['anomalyLabel']) self.assertEqual(state.setByUser, record['setByUser']) def mockRemoveIds(self, ids): self.helper.clamodel._getAnomalyClassifier().getSelf()._knn._numPatterns -= len(ids) for idx in ids: if idx in self.helper.clamodel._getAnomalyClassifier().getSelf().getParameter('categoryRecencyList'): self.helper.clamodel._getAnomalyClassifier().getSelf().getParameter('categoryRecencyList').remove(idx) if __name__ == '__main__': parser = TestOptionParser() options, args = parser.parse_args() # Form the command line for the unit test framework args = [sys.argv[0]] + args unittest.main(argv=args)
@unittest.skip("Currently Fails: NUP-1864") def test_TemporalAnomaly(self): """ Test that we get the same predictions out of a model that was saved and reloaded from a checkpoint as we do from one that runs continuously. """ self._testSamePredictions(experiment="temporal_anomaly", predSteps=1, checkpointAt=250, predictionsFilename='DefaultTask.TemporalAnomaly.predictionLog.csv', additionalFields=['anomalyScore']) @unittest.skip("We aren't currently supporting serialization backward " "compatibility") def test_BackwardsCompatibility(self): """ Test that we can load in a checkpoint saved by an earlier version of the OPF. """ self._testBackwardsCompatibility( os.path.join('backwards_compatibility', 'a'), 'savedmodels_2012-10-05') if __name__ == "__main__": initLogging(verbose=True) unittest.main()
"anomalyVector": "Vector", "anomalyLabel": "Label", "setByUser": False } state.__setstate__(record) self.assertEqual(state.ROWID, record['ROWID']) self.assertEqual(state.anomalyScore, record['anomalyScore']) self.assertEqual(state.anomalyVector, record['anomalyVector']) self.assertEqual(state.anomalyLabel, record['anomalyLabel']) self.assertEqual(state.setByUser, record['setByUser']) def mockRemoveIds(self, ids): self.helper.htm_prediction_model._getAnomalyClassifier().getSelf( )._knn._numPatterns -= len(ids) for idx in ids: if idx in self.helper.htm_prediction_model._getAnomalyClassifier( ).getSelf().getParameter('categoryRecencyList'): self.helper.htm_prediction_model._getAnomalyClassifier( ).getSelf().getParameter('categoryRecencyList').remove(idx) if __name__ == '__main__': parser = TestOptionParser() options, args = parser.parse_args() # Form the command line for the unit test framework args = [sys.argv[0]] + args unittest.main(argv=args)