Example #1
0
 def test_get_keyword_defaults(self):
     self.assertEqual(utils.get_keyword_defaults(self.f1), {})
     d = {'b': 1, 'c': 2, 'd': 3, 'e': 4}
     self.assertEqual(utils.get_keyword_defaults(self.f2), d)
     del d['b']
     del d['c']
     self.assertEqual(utils.get_keyword_defaults(self.f2, 2), d)
Example #2
0
 def test_get_keyword_defaults(self):
     self.assertEqual(utils.get_keyword_defaults(self.f1), {})
     d = {'b': 1, 'c': 2, 'd': 3, 'e': 4}
     self.assertEqual(utils.get_keyword_defaults(self.f2), d)
     del d['b']
     del d['c']
     self.assertEqual(utils.get_keyword_defaults(self.f2, 2), d)
Example #3
0
def test_get_keyword_defaults_f3():
    d = {'a': None, 'b': 1, 'c': 2, 'd': 3, 'e': 4}
    assert utils.get_keyword_defaults(f3) == d

    del d['a']
    assert utils.get_keyword_defaults(f3, 1) == d
    assert utils.get_keyword_defaults(f3, 7) == {}
Example #4
0
def test_get_keyword_defaults_f2():
    d = {'b': 1, 'c': 2, 'd': 3, 'e': 4}
    assert utils.get_keyword_defaults(f2) == d

    del d['b']
    del d['c']
    assert utils.get_keyword_defaults(f2, 2) == d
    assert utils.get_keyword_defaults(f2, 7) == {}
Example #5
0
 def __init__(self):
     self.priors = {}
     self._walker = Walk
     self._sampler = MetropolisMH
     self._sampler_opt = get_keyword_defaults(MetropolisMH.init)
     self.sample = None
     self.walk = lambda: None
     NoNewAttributesAfterInit.__init__(self)
Example #6
0
    def __init__(self):

        self.priors = {}

        # default sampler
        self.sampler = 'MetropolisMH'
        self.samplers = {
            'MH': {
                'class': MH,
                'opts': get_keyword_defaults(MH.init)
            },
            'MetropolisMH': {
                'class': MetropolisMH,
                'opts': get_keyword_defaults(MetropolisMH.init)
            },
        }

        self.walk = 'Walk'
        self.walks = {
            'Walk': {
                'class': Walk,
            },
        }
Example #7
0
    def __init__(self):

        self.priors = {}

        # default sampler
        self.sampler = 'MetropolisMH'
        self.samplers = {
            'MH' : {
                'class' : MH,
                'opts' : get_keyword_defaults(MH.init)
                },

            'MetropolisMH' : {
                'class' : MetropolisMH,
                'opts' : get_keyword_defaults(MetropolisMH.init)
                },
            }

        self.walk = 'Walk'
        self.walks = {
            'Walk' : {
                'class' : Walk,
                },
            }
Example #8
0
    def __setstate__(self, state):
        new_config = get_keyword_defaults(state.get('_optfunc'))
        old_config = state.get('config', {})

        # remove old kw args from opt method dict
        for key in old_config.keys():
            if key not in new_config:
                old_config.pop(key)

        # add new kw args with defaults
        for key, val in new_config.items():
            if key not in old_config:
                old_config[key] = val

        self.__dict__.update(state)
Example #9
0
    def __setstate__(self, state):
        new_config = get_keyword_defaults(state.get('_optfunc'))
        old_config = state.get('config', {})

        # remove old kw args from opt method dict
        for key in old_config.keys():
            if key not in new_config:
                old_config.pop(key)

        # add new kw args with defaults
        for key, val in new_config.items():
            if key not in old_config:
                old_config[key] = val

        self.__dict__.update(state)
Example #10
0
def test_get_keyword_defaults_f1():
    assert utils.get_keyword_defaults(f1) == {}
Example #11
0
def get_histo_defaults():
    return get_keyword_defaults(histo, 6)
Example #12
0
def get_contour_defaults():
    return get_keyword_defaults(contour, 6)
Example #13
0
def get_point_defaults():
    return get_keyword_defaults(point, 2)
Example #14
0
def get_plot_defaults():
    return get_keyword_defaults(plot, 7)
Example #15
0
def get_split_plot_defaults():
    return get_keyword_defaults(set_subplot, 3)
Example #16
0
def get_split_plot_defaults():
    return get_keyword_defaults(set_subplot, 1)
Example #17
0
 def _set_sampler(self, sampler):
     self._sampler = sampler
     self._sampler_opt = get_keyword_defaults(sampler.init)
Example #18
0
 def _get_default_config(self):
     args = get_keyword_defaults(self._optfunc)
     return args
Example #19
0
def get_histo_defaults():
    return get_keyword_defaults(_histogram, 6)
Example #20
0
def get_histo_defaults():
    return get_keyword_defaults(histo, 6)
Example #21
0
def get_point_defaults():
    return get_keyword_defaults(point, 2)
Example #22
0
def get_plot_defaults():
    return get_keyword_defaults(plot, 7)
Example #23
0
def get_histo_defaults():
    return get_keyword_defaults(_histogram, 6)
Example #24
0
        pylab.draw()

def exceptions():
    pass

def _choose(test, iftrue, iffalse=None):
    if test:
        return iftrue
    return iffalse


# In matplotlib 1.5RC1 the kwargs to pylab.Axes.errorbar are not
# explictit, but they appear to be set for pybal.errorbar, so
# switch to that.
# _errorbar_defaults = get_keyword_defaults(pylab.Axes.errorbar)
_errorbar_defaults = get_keyword_defaults(pylab.errorbar)


def clear_window():
    pylab.clf()

def set_window_redraw(redraw):
    if redraw:
        pylab.draw()


def point(x, y, overplot=True, clearwindow=False,
          symbol=None,
          color=None):

    if overplot:
Example #25
0
def get_contour_defaults():
    return get_keyword_defaults(contour, 6)
Example #26
0
def begin():
    pass


def end():
    set_window_redraw(True)
    if plt.isinteractive():
        plt.draw()


def exceptions():
    pass


_errorbar_defaults = get_keyword_defaults(plt.errorbar)


def clear_window():
    plt.clf()


def set_window_redraw(redraw):
    if redraw:
        plt.draw()


def setup_axes(overplot, clearwindow):
    """Return the axes object, creating it if necessary.

    Parameters
Example #27
0
    def _get_default_config(self):
        args = get_keyword_defaults(self._optfunc)
	return args
Example #28
0
def exceptions():
    pass


def _choose(test, iftrue, iffalse=None):
    if test:
        return iftrue
    return iffalse


# In matplotlib 1.5RC1 the kwargs to pylab.Axes.errorbar are not
# explictit, but they appear to be set for pybal.errorbar, so
# switch to that.
# _errorbar_defaults = get_keyword_defaults(pylab.Axes.errorbar)
_errorbar_defaults = get_keyword_defaults(pylab.errorbar)


def clear_window():
    pylab.clf()


def set_window_redraw(redraw):
    if redraw:
        pylab.draw()


def point(x, y, overplot=True, clearwindow=False, symbol=None, color=None):

    if overplot:
        axes = pylab.gca()