Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use draw_symbol method in hypothesis

Best Python code snippet using hypothesis

ap.py

Source:ap.py

1"""2Package that allows you to plot simple graphs in ASCII, a la matplotlib.3This package is a inspired from Imri Goldberg's ASCII-Plotter 1.04(https://pypi.python.org/pypi/ASCII-Plotter/1.0)5At a time I was enoyed by security not giving me direct access to my computer,6and thus to quickly make figures from python, I looked at how I could make7quick and dirty ASCII figures. But if I were to develop something, I wanted8something that can be used with just python and possible standard-ish packages9(numpy, scipy).10So I came up with this package after many iterations based of ASCII-plotter.11I added the feature to show multiple curves on one plot with different markers.12And I also made the usage, close to matplotlib, such that there is a plot,13hist, hist2d and imshow functions.14TODO:15    imshow does not plot axis yet.16    make a correct documentation17"""18import math as _math19import numpy as np20__version__ = 0.921__author__ = 'M. Fouesneau'22__all__ = ['markers', 'ACanvas', 'AData', 'AFigure',23           'hist', 'hist2d', 'imshow', 'percentile_imshow',24           'plot', 'stem', 'stemify', 'step', 'steppify',25           '__version__', '__author__']26markers = { '-' : u'None' ,     # solid line style27            ',': u'\u2219',     # point marker28            '.': u'\u2218',     # pixel marker29            '.f': u'\u2218',    # pixel marker30            'o': u'\u25CB',     # circle marker31            'of': u'\u25CF',    # circle marker32            'v': u'\u25BD',     # triangle_down marker33            'vf': u'\u25BC',    # filler triangle_down marker34            '^': u'\u25B3',     # triangle_up marker35            '^f': u'\u25B2',    # filled triangle_up marker36            '<':  u'\u25C1',    # triangle_left marker37            '<f': u'\u25C0',    # filled triangle_left marker38            '>': u'\u25B7',     # triangle_right marker39            '>f': u'\u25B6',    # filled triangle_right marker40            's': u'\u25FD',     # square marker41            'sf': u'\u25FC',    # square marker42            '*': u'\u2606',     # star marker43            '*f': u'\u2605',    # star marker44            '+': u'\u271A',     # plus marker45            'x': u'\u274C',     # x marker46            'd':  u'\u25C7',    # diamond marker47            'df':  u'\u25C6'    # filled diamond marker48            }49def _sign(x):50    """ Return the sign of x51    INPUTS52    ------53    x: number54        value to get the sign of55    OUTPUTS56    -------57    s: signed int58        -1, 0 or 1 if negative, null or positive59    """60    if (x > 0):61        return 162    elif (x == 0):63        return 064    else:65        return -166def _transpose(mat):67    """ Transpose matrice made of lists68    INPUTS69    ------70    mat: iterable 2d list like71    OUTPUTS72    -------73    r: list of list, 2d list like74        transposed matrice75    """76    r = [ [x[i] for x in mat] for i in range(len(mat[0])) ]77    return r78def _y_reverse(mat):79    """ Reverse the y axis of a 2d list-like80    INPUTS81    ------82    mat: list of lists83        the matrix to reverse on axis 084    OUTPUTS85    -------86    r: list of lists87        the reversed version88    """89    r = [ list(reversed(mat_i)) for mat_i in mat ]90    return r91class AData(object):92    """ Data container for ascii AFigure """93    def __init__(self, x, y, marker='_.', plot_slope=True):94        """ Constructor95        INPUTS96        ------97        x: iterable98            x values99        y: iterable100            y values101        KEYWORDS102        --------103        marker: str104            marker for the data.105            if None or empty, the curve will be plotted106            if the first character of the marker is '_' then unicode markers will be called:107                marker      repr                description108               ========  ===========   =============================109                '-'       u'None'       solid line style110                ','       u'\u2219'     point marker111                '.'       u'\u2218'     pixel marker112                '.f'      u'\u2218'     pixel marker113                'o'       u'\u25CB'     circle marker114                'of'      u'\u25CF'     circle marker115                'v'       u'\u25BD'     triangle_down marker116                'vf'      u'\u25BC'     filler triangle_down marker117                '^'       u'\u25B3'     triangle_up marker118                '^f'      u'\u25B2'     filled triangle_up marker119                '<'       u'\u25C1'     triangle_left marker120                '<f'      u'\u25C0'     filled triangle_left marker121                '>'       u'\u25B7'     triangle_right marker122                '>f'      u'\u25B6'     filled triangle_right marker123                's'       u'\u25FD'     square marker124                'sf'      u'\u25FC'     square marker125                '*'       u'\u2606'     star marker126                '*f'      u'\u2605'     star marker127                '+'       u'\u271A'     plus marker128                'x'       u'\u274C'     x marker129                'd'       u'\u25C7'     diamond marker130                'df'      u'\u25C6'     filled diamond marker131        plot_slope: bool132            if set, the curve will be plotted133        """134        self.x = x135        self.y = y136        self.plot_slope = plot_slope137        self.set_marker(marker)138    def set_marker(self, marker):139        """ set the marker of the data140        INPUTS141        ------142        marker: str143            marker for the data.144            see constructor for marker descriptions145        """146        if marker in [None, 'None', u'None', '']:147            self.plot_slope = True148            self.marker = ''149        elif marker[0] == '_':150            self.marker = markers[marker[1:]]151        else:152            self.marker = marker153    def extent(self):154        """ return the extention of the data155        OUPUTS156        ------157        e: list158           [ min(x), max(x), min(y), max(y) ]159        """160        return [min(self.x), max(self.x), min(self.y), max(self.y)]161    def __repr__(self):162        s = 'AData: %s\n' % object.__repr__(self)163        return s164class ACanvas(object):165    """ Canvas of a AFigure instance. A Canvas handles all transformations166    between data space and figure space accounting for scaling and pixels167    In general there is no need to access the canvas directly168    """169    def __init__(self, shape=None, margins=None, xlim=None, ylim=None):170        """ Constructor171        KEYWORDS172        --------173        shape: tuple of 2 ints174            shape of the canvas in number of characters: (width, height)175        margins: tuple of 2 floats176            fractional margins177        xlim: tuple of 2 floats178            limits of the xaxis179        ylim: tuple of 2 floats180            limits of the yaxis181        """182        self.shape = shape or (50, 20)183        self.margins = margins or (0.05, 0.1)184        self._xlim = xlim or [0, 1]185        self._ylim = ylim or [0, 1]186        self.auto_adjust = True187        self.margin_factor = 1188    @property189    def x_size(self):190        """ return the width """191        return self.shape[0]192    @property193    def y_size(self):194        """ return the height """195        return self.shape[1]196    @property197    def x_margin(self):198        """ return the margin in x """199        return self.margins[0]200    @property201    def y_margin(self):202        """ return the margin in y """203        return self.margins[1]204    def xlim(self, vmin=None, vmax=None):205        """206        Get or set the *x* limits of the current axes.207        KEYWORDS208        --------209        vmin: float210            lower limit211        vmax: float212            upper limit213        xmin, xmax = xlim()   # return the current xlim214        xlim( (xmin, xmax) )  # set the xlim to xmin, xmax215        xlim( xmin, xmax )    # set the xlim to xmin, xmax216        """217        if vmin is None and vmax is None:218            return self._xlim219        elif hasattr(vmin, '__iter__'):220            self._xlim = vmin[:2]221        else:222            self._xlim = [vmin, vmax]223        if self._xlim[0] == self._xlim[1]:224            self._xlim[1] += 1225        self._xlim[0] -= self.x_mod226        self._xlim[1] += self.x_mod227    def ylim(self, vmin=None, vmax=None):228        """229        Get or set the *y* limits of the current axes.230        KEYWORDS231        --------232        vmin: float233            lower limit234        vmax: float235            upper limit236        ymin, ymax = ylim()   # return the current xlim237        ylim( (ymin, ymax) )  # set the xlim to xmin, xmax238        ylim( ymin, ymax )    # set the xlim to xmin, xmax239        """240        if vmin is None and vmax is None:241            return self._ylim242        elif hasattr(vmin, '__iter__'):243            self._ylim = vmin[:2]244        else:245            self._ylim = [vmin, vmax]246        if self._ylim[0] == self._ylim[1]:247            self._ylim[1] += 1248        self._ylim[0] -= self.y_mod249        self._ylim[1] += self.y_mod250    @property251    def min_x(self):252        """ return the lower x limit """253        return self._xlim[0]254    @property255    def max_x(self):256        """ return the upper x limit """257        return self._xlim[1]258    @property259    def min_y(self):260        """ return the lower y limit """261        return self._ylim[0]262    @property263    def max_y(self):264        """ return the upper y limit """265        return self._ylim[1]266    @property267    def x_step(self):268        return float(self.max_x - self.min_x) / float(self.x_size)269    @property270    def y_step(self):271        return float(self.max_y - self.min_y) / float(self.y_size)272    @property273    def ratio(self):274        return self.y_step / self.x_step275    @property276    def x_mod(self):277        return (self.max_x - self.min_x) * self.x_margin278    @property279    def y_mod(self):280        return (self.max_y - self.min_y) * self.y_margin281    def extent(self, margin_factor=None):282        margin_factor = margin_factor or self.margin_factor283        min_x = (self.min_x + self.x_mod * margin_factor)284        max_x = (self.max_x - self.x_mod * margin_factor)285        min_y = (self.min_y + self.y_mod * margin_factor)286        max_y = (self.max_y - self.y_mod * margin_factor)287        return (min_x, max_x, min_y, max_y)288    def extent_str(self, margin=None):289        def transform(val, fmt):290            if abs(val) < 1:291                _str = "%+.2g" % val292            elif fmt is not None:293                _str = fmt % val294            else:295                _str = None296            return _str297        e = self.extent(margin)298        xfmt = self.x_str()299        yfmt = self.y_str()300        return transform(e[0], xfmt), transform(e[1], xfmt), transform(e[2], yfmt), transform(e[3], yfmt)301    def x_str(self):302        if self.x_size < 16:303            x_str = None304        elif self.x_size < 23:305            x_str = "%+.2g"306        else:307            x_str = "%+g"308        return x_str309    def y_str(self):310        if self.x_size < 8:311            y_str = None312        elif self.x_size < 11:313            y_str = "%+.2g"314        else:315            y_str = "%+g"316        return y_str317    def coords_inside_buffer(self, x, y):318        return (0 <= x < self.x_size) and (0 < y < self.y_size)319    def coords_inside_data(self, x, y):320        """ return if (x,y) covered by the data box321        x, y: float322            coordinates to test323        """324        return (self.min_x <= x < self.max_x) and (self.min_y <= y < self.max_y)325    def _clip_line(self, line_pt_1, line_pt_2):326        """ clip a line to the canvas """327        e = self.extent()328        x_min = min(line_pt_1[0], line_pt_2[0])329        x_max = max(line_pt_1[0], line_pt_2[0])330        y_min = min(line_pt_1[1], line_pt_2[1])331        y_max = max(line_pt_1[1], line_pt_2[1])332        if line_pt_1[0] == line_pt_2[0]:333            return ( ( line_pt_1[0], max(y_min, e[1]) ),334                     ( line_pt_1[0], min(y_max, e[3])   ))335        if line_pt_1[1] == line_pt_2[1]:336            return ( ( max(x_min, e[0]), line_pt_1[1] ),337                     ( min(x_max, e[2]), line_pt_1[1]   ))338        if ( (e[0] <= line_pt_1[0] < e[2]) and339             (e[1] <= line_pt_1[1] < e[3]) and340             (e[0] <= line_pt_2[0] < e[2]) and341             (e[1] <= line_pt_2[1] < e[3]) ):342            return line_pt_1, line_pt_2343        ts = [0.0,344              1.0,345              float(e[0] - line_pt_1[0]) / (line_pt_2[0] - line_pt_1[0]),346              float(e[2] - line_pt_1[0]) / (line_pt_2[0] - line_pt_1[0]),347              float(e[1] - line_pt_1[1]) / (line_pt_2[1] - line_pt_1[1]),348              float(e[3] - line_pt_1[1]) / (line_pt_2[1] - line_pt_1[1])349              ]350        ts.sort()351        if (ts[2] < 0) or (ts[2] >= 1) or (ts[3] < 0) or (ts[2] >= 1):352            return None353        result = [(pt_1 + t * (pt_2 - pt_1)) for t in (ts[2], ts[3]) for (pt_1, pt_2) in zip(line_pt_1, line_pt_2)]354        return ( result[:2], result[2:] )355class AFigure(object):356    def __init__(self, shape=(80, 20), margins=(0.05, 0.1), draw_axes=True, newline='\n',357                 plot_labels=True, xlim=None, ylim=None, **kwargs):358        self.canvas = ACanvas(shape, margins=margins, xlim=xlim, ylim=ylim)359        self.draw_axes = draw_axes360        self.new_line = newline361        self.plot_labels = plot_labels362        self.output_buffer = None363        self.tickSymbols = u'\u253C'  # "+"364        self.x_axis_symbol = u'\u2500'  # u"\u23bc"  # "-"365        self.y_axis_symbol = u'\u2502'  # "|"366        self.data = []367    def xlim(self, vmin=None, vmax=None):368        return self.canvas.xlim(vmin, vmax)369    def ylim(self, vmin=None, vmax=None):370        return self.canvas.ylim(vmin, vmax)371    def get_coord(self, val, min, step, limits=None):372        result = int((val - min) / step)373        if limits is not None:374            if result <= limits[0]:375                result = limits[0]376            elif result >= limits[1]:377                result = limits[1] - 1378        return result379    def _draw_axes(self):380        zero_x = self.get_coord(0, self.canvas.min_x, self.canvas.x_step, limits=[1, self.canvas.x_size])381        if zero_x >= self.canvas.x_size:382            zero_x = self.canvas.x_size - 1383        for y in range(self.canvas.y_size):384            self.output_buffer[zero_x][y] = self.y_axis_symbol385        zero_y = self.get_coord(0, self.canvas.min_y, self.canvas.y_step, limits=[1, self.canvas.y_size])386        if zero_y >= self.canvas.y_size:387            zero_y = self.canvas.y_size - 1388        for x in range(self.canvas.x_size):389            self.output_buffer[x][zero_y] = self.x_axis_symbol  # u'\u23bc'390        self.output_buffer[zero_x][zero_y] = self.tickSymbols  # "+"391    def _get_symbol_by_slope(self, slope, default_symbol):392        """ Return a line oriented directed approximatively along the slope value """393        if slope > _math.tan(3 * _math.pi / 8):394            draw_symbol = "|"395        elif _math.tan(_math.pi / 8) < slope < _math.tan(3 * _math.pi / 8):396            draw_symbol = u'\u27cb'  # "/"397        elif abs(slope) < _math.tan(_math.pi / 8):398            draw_symbol = "-"399        elif slope < _math.tan(-_math.pi / 8) and slope > _math.tan(-3 * _math.pi / 8):400            draw_symbol = u'\u27CD'  # "\\"401        elif slope < _math.tan(-3 * _math.pi / 8):402            draw_symbol = "|"403        else:404            draw_symbol = default_symbol405        return draw_symbol406    def _plot_labels(self):407        if self.canvas.y_size < 2:408            return409        act_min_x, act_max_x, act_min_y, act_max_y = self.canvas.extent()410        min_x_coord = self.get_coord(act_min_x, self.canvas.min_x, self.canvas.x_step, limits=[0, self.canvas.x_size])411        max_x_coord = self.get_coord(act_max_x, self.canvas.min_x, self.canvas.x_step, limits=[0, self.canvas.x_size])412        min_y_coord = self.get_coord(act_min_y, self.canvas.min_y, self.canvas.y_step, limits=[1, self.canvas.y_size])413        max_y_coord = self.get_coord(act_max_y, self.canvas.min_y, self.canvas.y_step, limits=[1, self.canvas.y_size])414        x_zero_coord = self.get_coord(0, self.canvas.min_x, self.canvas.x_step, limits=[0, self.canvas.x_size])415        y_zero_coord = self.get_coord(0, self.canvas.min_y, self.canvas.y_step, limits=[1, self.canvas.y_size])416        self.output_buffer[x_zero_coord][min_y_coord] = self.tickSymbols417        self.output_buffer[x_zero_coord][max_y_coord] = self.tickSymbols418        self.output_buffer[min_x_coord][y_zero_coord] = self.tickSymbols419        self.output_buffer[max_x_coord][y_zero_coord] = self.tickSymbols420        min_x_str, max_x_str, min_y_str, max_y_str = self.canvas.extent_str()421        if (self.canvas.x_str() is not None):422            for i, c in enumerate(min_x_str):423                self.output_buffer[min_x_coord + i + 1][y_zero_coord - 1] = c424            for i, c in enumerate(max_x_str):425                self.output_buffer[max_x_coord + i - len(max_x_str)][y_zero_coord - 1] = c426        if (self.canvas.y_str() is not None):427            for i, c in enumerate(max_y_str):428                self.output_buffer[x_zero_coord + i + 1][max_y_coord] = c429            for i, c in enumerate(min_y_str):430                self.output_buffer[x_zero_coord + i + 1][min_y_coord] = c431    def _plot_line(self, start, end, data):432        """ plot a line from start = (x0, y0) to end = (x1, y1) """433        clipped_line = self.canvas._clip_line(start, end)434        if clipped_line is None:435            return False436        start, end = clipped_line437        x0 = self.get_coord(start[0], self.canvas.min_x, self.canvas.x_step)438        y0 = self.get_coord(start[1], self.canvas.min_y, self.canvas.y_step)439        x1 = self.get_coord(end[0], self.canvas.min_x, self.canvas.x_step)440        y1 = self.get_coord(end[1], self.canvas.min_y, self.canvas.y_step)441        if (x0, y0) == (x1, y1):442            return True443        #x_zero_coord = self.get_coord(0, self.canvas.min_x, self.canvas.x_step)444        y_zero_coord = self.get_coord(0, self.canvas.min_y, self.canvas.y_step, limits=[1, self.canvas.y_size])445        if start[0] - end[0] == 0:446            draw_symbol = u"|"447        elif start[1] - end[1] == 0:448            draw_symbol = '-'449        else:450            slope = (1.0 / self.canvas.ratio) * (end[1] - start[1]) / (end[0] - start[0])451            draw_symbol = self._get_symbol_by_slope(slope, data.marker)452        dx = x1 - x0453        dy = y1 - y0454        if abs(dx) > abs(dy):455            s = _sign(dx)456            slope = float(dy) / dx457            for i in range(0, abs(int(dx))):458                cur_draw_symbol = draw_symbol459                x = i * s460                cur_y = int(y0 + slope * x)461                if (self.draw_axes) and (cur_y == y_zero_coord) and (draw_symbol == self.x_axis_symbol):462                    cur_draw_symbol = "-"463                self.output_buffer[x0 + x][cur_y] = cur_draw_symbol464        else:465            s = _sign(dy)466            slope = float(dx) / dy467            for i in range(0, abs(int(dy))):468                y = i * s469                cur_draw_symbol = draw_symbol470                cur_y = y0 + y471                if (self.draw_axes) and (cur_y == y_zero_coord) and (draw_symbol == self.x_axis_symbol):472                    cur_draw_symbol = "-"473                self.output_buffer[int(x0 + slope * y)][cur_y] = cur_draw_symbol474        return False475    def _plot_data_with_slope(self, data):476        xy = list(zip(data.x, data.y))477        #sort according to the x coord478        xy.sort(key=lambda c: c[0])479        prev_p = xy[0]480        e_xy = enumerate(xy)481        e_xy.next()482        for i, (xi, yi) in e_xy:483            line = self._plot_line(prev_p, (xi, yi), data)484            prev_p = (xi, yi)485            # if no line, then symbol486            if not line & self.canvas.coords_inside_data(xi, yi):487                draw_symbol = data.marker488                px, py = xy[i - 1]489                nx, ny = xy[i]490                if abs(nx - px) > 0.000001:491                    slope = (1.0 / self.canvas.ratio) * (ny - py) / (nx - px)492                    draw_symbol = self._get_symbol_by_slope(slope, draw_symbol)493                x_coord = self.get_coord(xi, self.canvas.min_x, self.canvas.x_step)494                y_coord = self.get_coord(yi, self.canvas.min_y, self.canvas.y_step)495                if self.canvas.coords_inside_buffer(x_coord, y_coord):496                    y0_coord = self.get_coord(0, self.canvas.min_y, self.canvas.y_step)497                    if self.draw_axes:498                        if (y_coord == y0_coord) and (draw_symbol == u"\u23bc"):499                            draw_symbol = "="500                    self.output_buffer[x_coord][y_coord] = draw_symbol501    def _plot_data(self, data):502        if data.plot_slope:503            self._plot_data_with_slope(data)504        else:505            for x, y in zip(data.x, data.y):506                if self.canvas.coords_inside_data(x, y):507                    x_coord = self.get_coord(x, self.canvas.min_x, self.canvas.x_step)508                    y_coord = self.get_coord(y, self.canvas.min_y, self.canvas.y_step)509                    if self.canvas.coords_inside_buffer(x_coord, y_coord):510                        self.output_buffer[x_coord][y_coord] = data.marker511    def auto_limits(self):512        if self.canvas.auto_adjust is True:513            min_x = 0.514            max_x = 0.515            min_y = 0.516            max_y = 0.517            for dk in self.data:518                ek = dk.extent()519                min_x = min(min_x, min(ek[:2]))520                min_y = min(min_y, min(ek[2:]))521                max_x = max(max_x, max(ek[:2]))522                max_y = max(max_y, max(ek[2:]))523            self.canvas.xlim(min_x, max_x)524            self.canvas.ylim(min_y, max_y)525    def append_data(self, data):526        self.data.append(data)527        self.auto_limits()528    def plot(self, x_seq, y_seq=None, marker=None, plot_slope=False, xlim=None, ylim=None):529        if y_seq is None:530            y_seq = x_seq[:]531            x_seq = range(len(y_seq))532        data = AData(x_seq, y_seq, marker=marker, plot_slope=plot_slope)533        self.append_data(data)534        if xlim is not None:535            self.canvas.xlim(xlim)536        if ylim is not None:537            self.canvas.ylim(xlim)538        return self.draw()539    def draw(self):540        self.output_buffer = [[" "] * self.canvas.y_size for i in range(self.canvas.x_size)]541        if self.draw_axes:542            self._draw_axes()543        for dk in self.data:544            self._plot_data(dk)545        if self.plot_labels:546            self._plot_labels()547        trans_result = _transpose(_y_reverse(self.output_buffer))548        result = self.new_line.join(["".join(row) for row in trans_result])549        return result550def plot(x, y=None, marker=None, shape=(50, 20), draw_axes=True,551         newline='\n', plot_slope=False, x_margin=0.05,552         y_margin=0.1, plot_labels=True, xlim=None, ylim=None):553    flags = {'shape': shape,554             'draw_axes': draw_axes,555             'newline': newline,556             'marker': marker,557             'plot_slope': plot_slope,558             'margins': (x_margin, y_margin),559             'plot_labels': plot_labels }560    p = AFigure(**flags)561    print(p.plot(x, y, marker=marker, plot_slope=plot_slope))562def steppify(x, y):563    """ Steppify a curve (x,y). Useful for manually filling histograms """564    dx = 0.5 * (x[1:] + x[:-1])565    xx = np.zeros( 2 * len(dx), dtype=float)566    yy = np.zeros( 2 * len(y), dtype=float)567    xx[0::2], xx[1::2] = dx, dx568    yy[0::2], yy[1::2] = y, y569    xx = np.concatenate(([x[0] - (dx[0] - x[0])], xx, [x[-1] + (x[-1] - dx[-1])]))570    return xx, yy571def stemify(x, y):572    """ Steppify a curve (x,y). Useful for manually filling histograms """573    xx = np.zeros( 3 * len(x), dtype=float)574    yy = np.zeros( 3 * len(y), dtype=float)575    xx[0::3], xx[1::3], xx[2::3] = x, x, x576    yy[1::3] = y577    return xx, yy578def hist(x, bins=10, normed=False, weights=None, density=None, histtype='stem',579         shape=(50, 20), draw_axes=True, newline='\n',580         marker='_.', plot_slope=False, x_margin=0.05,581         y_margin=0.1, plot_labels=True, xlim=None, ylim=None ):582    from numpy import histogram583    if histtype not in ['None', 'stem', 'step']:584        raise ValueError('histtype must be in [None, stem, step]')585    n, b = histogram(x, bins=bins, range=xlim, normed=normed, weights=weights, density=density)586    _x = 0.5 * ( b[:-1] + b[1:] )587    if histtype == 'step':588        step(_x, n.astype(float))589    elif histtype == 'stem':590        stem(_x, n.astype(float))591    else:592        _y = n.astype(float)593        plot(_x, _y, shape=shape, draw_axes=draw_axes, newline=newline, marker=marker,594             plot_slope=plot_slope, x_margin=x_margin, y_margin=y_margin,595             plot_labels=plot_labels, xlim=xlim, ylim=ylim)596def step(x, y, shape=(50, 20), draw_axes=True,597         newline='\n', marker='_.', plot_slope=True, x_margin=0.05,598         y_margin=0.1, plot_labels=True, xlim=None, ylim=None ):599    _x, _y = steppify(x, y)600    plot(_x, _y, shape=shape, draw_axes=draw_axes, newline=newline, marker=marker,601         plot_slope=plot_slope, x_margin=x_margin, y_margin=y_margin,602         plot_labels=plot_labels, xlim=xlim, ylim=ylim)603def stem(x, y, shape=(50, 20), draw_axes=True,604         newline='\n', marker='_.', plot_slope=True, x_margin=0.05,605         y_margin=0.1, plot_labels=True, xlim=None, ylim=None ):606    _x, _y = stemify(x, y)607    plot(_x, _y, shape=shape, draw_axes=draw_axes, newline=newline, marker=marker,608         plot_slope=plot_slope, x_margin=x_margin, y_margin=y_margin,609         plot_labels=plot_labels, xlim=xlim, ylim=ylim)610def hist2d(x, y, bins=[50, 20], range=None, normed=False, weights=None, ncolors=16,611           width=50, percentiles=None):612    im, ex, ey = np.histogram2d(x, y, bins, range=None, normed=normed, weights=weights)613    if percentiles is None:614        imshow(im, extent=[min(ex), max(ex), min(ey), max(ey)],615               ncolors=ncolors, width=width)616    else:617        percentile_imshow(im, levels=percentiles, extent=None,618                          width=width, ncolors=width)619def percentile_imshow(im, levels=[68, 95, 99], extent=None, width=50, ncolors=16):620    _im = im.astype(float)621    _im -= im.min()622    _im /= _im.max()623    n = len(levels)624    for e, lk in enumerate(sorted(levels)):625        _im[ _im <= 0.01 * float(lk) ] = n - e626    imshow(1. - _im, extent=None, width=width, ncolors=ncolors)627def imshow(im, extent=None, width=50, ncolors=16):628    from scipy import ndimage629    width0 = im.shape[0]630    _im = ndimage.zoom(im.astype(float), float(width) / float(width0) )631    _im -= im.min()632    _im /= _im.max()633    width, height = _im.shape[:2]634    if len(im.shape) > 2:635        _clr = True636    else:637        _clr = False638    if ncolors == 16:639        color = "MNHQ$OC?7>!:-;. "[::-1]640    else:641        color = '''$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,"^`'. '''[::-1]642    ncolors = len(color)643    string = ""644    if not _clr:645        for h in range(height):  # first go through the height,  otherwise will roate646            for w in range(width):647                string += color[int(_im[w, h] * (ncolors - 1) )]648            string += "\n"649    else:650        for h in range(height):  # first go through the height,  otherwise will roate651            for w in range(width):652                string += color[int(sum(_im[w, h]) * (ncolors - 1) )]653            string += "\n"...

opencv_filter.py

Source:opencv_filter.py

...222    output = np.dstack([frame, np.ones((h, w), dtype="uint8") * 255])223    # Threshold the HSV image224    # Red225    light_red_mask = cv2.inRange(hsv, lower_light_red, upper_light_red)226    draw_symbol(light_red_mask, detector, 'vermelho.png', output, h, w)227    228    red_mask = cv2.inRange(hsv, lower_red, upper_red)229    draw_symbol(red_mask, detector, 'vermelho.png', output, h, w)230    231    dark_red_mask = cv2.inRange(hsv, lower_dark_red, upper_dark_red)232    draw_symbol(dark_red_mask, detector, 'vermelho.png', output, h, w)233    # Yellow234    light_yellow_mask = cv2.inRange(hsv, lower_light_yellow, upper_light_yellow)235    draw_symbol(light_yellow_mask, detector, 'amarelo.png', output, h, w)236    237    yellow_mask = cv2.inRange(hsv, lower_yellow, upper_yellow)238    draw_symbol(yellow_mask, detector, 'amarelo.png', output, h, w)239    240    dark_yellow_mask = cv2.inRange(hsv, lower_dark_yellow, upper_dark_yellow)241    draw_symbol(dark_yellow_mask, detector, 'amarelo.png', output, h, w)242    # Orange243    light_orange_mask = cv2.inRange(hsv, lower_light_orange, upper_light_orange)244    draw_symbol(light_orange_mask, detector, 'laranja.png', output, h, w)245    246    orange_mask = cv2.inRange(hsv, lower_orange, upper_orange)247    draw_symbol(orange_mask, detector, 'laranja.png', output, h, w)248    249    dark_orange_mask = cv2.inRange(hsv, lower_dark_orange, upper_dark_orange)250    draw_symbol(dark_orange_mask, detector, 'laranja.png', output, h, w)251    # Brown252    light_brown_mask = cv2.inRange(hsv, lower_light_brown, upper_light_brown)253    draw_symbol(light_brown_mask, detector, 'castanho.png', output, h, w)254    255    brown_mask = cv2.inRange(hsv, lower_brown, upper_brown)256    draw_symbol(brown_mask, detector, 'castanho.png', output, h, w)257    258    dark_brown_mask = cv2.inRange(hsv, lower_dark_brown, upper_dark_brown)259    draw_symbol(dark_brown_mask, detector, 'castanho.png', output, h, w)260    # Green261    light_green_mask = cv2.inRange(hsv, lower_light_green, upper_light_green)262    draw_symbol(light_green_mask, detector, 'verde.png', output, h, w)263    264    green_mask = cv2.inRange(hsv, lower_green, upper_green)265    draw_symbol(green_mask, detector, 'verde.png', output, h, w)266    267    dark_green_mask = cv2.inRange(hsv, lower_dark_green, upper_dark_green)268    draw_symbol(dark_green_mask, detector, 'verde.png', output, h, w)269    # Blue270    light_blue_mask = cv2.inRange(hsv, lower_light_blue, upper_light_blue)271    draw_symbol(light_blue_mask, detector, 'azul.png', output, h, w)272    blue_mask = cv2.inRange(hsv, lower_blue, upper_blue)273    draw_symbol(blue_mask, detector, 'azul.png', output, h, w)274    dark_blue_mask = cv2.inRange(hsv, lower_dark_blue, upper_dark_blue)275    draw_symbol(dark_blue_mask, detector, 'azul.png', output, h, w)276    # Violet277    light_violet_mask = cv2.inRange(hsv, lower_light_violet, upper_light_violet)278    draw_symbol(light_violet_mask, detector, 'violeta.png', output, h, w)279    280    violet_mask = cv2.inRange(hsv, lower_violet, upper_violet)281    draw_symbol(violet_mask, detector, 'violeta.png', output, h, w)282    dark_violet_mask = cv2.inRange(hsv, lower_dark_violet, upper_dark_violet)283    draw_symbol(dark_violet_mask, detector, 'violeta.png', output, h, w)284    # Black285    black_mask = cv2.inRange(hsv, lower_black, upper_black)286    white_mask = cv2.inRange(hsv, lower_white, upper_white)287  288    ratio_blue = getPercentage(frame, hsv, lower_dark_blue, upper_blue)289    ratio_blue = ratio_blue + getPercentage(frame, hsv, lower_light_blue, upper_light_blue)290    ratio_blue = ratio_blue + getPercentage(frame, hsv, lower_dark_blue, upper_dark_blue)291    log.update({'blue' : ratio_blue})292    ratio_green = getPercentage(frame, hsv, lower_dark_green, upper_green)293    ratio_green = ratio_green + getPercentage(frame, hsv, lower_light_green, upper_light_green)294    ratio_green = ratio_green + getPercentage(frame, hsv, lower_dark_green, upper_dark_green)295    log.update({'green' : ratio_green})296    ratio_yellow = getPercentage(frame, hsv, lower_dark_yellow, upper_yellow)297    ratio_yellow = ratio_yellow + getPercentage(frame, hsv, lower_light_yellow, upper_light_yellow)298    ratio_yellow = ratio_yellow + getPercentage(frame, hsv, lower_dark_yellow, upper_dark_yellow)299    log.update({'yellow' : ratio_yellow})300    ratio_orange = getPercentage(frame, hsv, lower_dark_orange, upper_orange)301    ratio_orange = ratio_orange + getPercentage(frame, hsv, lower_light_orange, upper_light_orange)302    ratio_orange = ratio_orange + getPercentage(frame, hsv, lower_dark_orange, upper_dark_orange)303    log.update({'orange' : ratio_orange})304    ratio_red = getPercentage(frame, hsv, lower_dark_red, upper_red)305    ratio_red = ratio_red + getPercentage(frame, hsv, lower_light_red, upper_light_red)306    ratio_red = ratio_red +  getPercentage(frame, hsv, lower_dark_red, upper_dark_red)307    log.update({'red' : ratio_red})308    ratio_violet = getPercentage(frame, hsv, lower_dark_violet, upper_violet)309    ratio_violet = ratio_violet + getPercentage(frame, hsv, lower_light_violet, upper_light_violet)310    ratio_violet = ratio_violet + getPercentage(frame, hsv, lower_dark_violet, upper_dark_violet)311    log.update({'violet' : ratio_violet})312    ratio_brown = getPercentage(frame, hsv, lower_dark_brown, upper_brown)313    ratio_brown = ratio_brown + getPercentage(frame, hsv, lower_light_brown, upper_light_brown)314    ratio_brown = ratio_brown + getPercentage(frame, hsv, lower_dark_brown, upper_dark_brown)315    log.update({'brown' : ratio_brown})316    ratio_white = getPercentage(frame, hsv, lower_white, upper_white)317    log.update({'white' : ratio_white})318    ratio_black = getPercentage(frame, hsv, lower_black, upper_black)319    log.update({'black' : ratio_black})320  321    with open("ratio.csv", "a") as write_file:322        write_file.write("\n")323        write_file.close()324    print(log)325    return output, log326def draw_symbol(color_mask, detector, file_name, output, h, w):327    #print(file_name)328    # Detect blobs.329    mask = cv2.bitwise_not(color_mask)330    keypoints = detector.detect(mask)331    332    # Load symbol333    watermark = cv2.imread("symbols/"+file_name, -1)334    watermark = cv2.resize(watermark, (20,20))335    (wH, wW) = watermark.shape[:2]336    #print(type(keypoints))337    #print(keypoints)338    for point in keypoints:339        #print(str(h)+" "+str(w))340        overlay = np.zeros((h, w, 4), dtype="uint8")...

minimalgame.py

Source:minimalgame.py

...22    23    def	game_loop(self):24		# this gets executed every frame25        if self.key_up():26            self.draw_symbol(1,1, NamedSymbol.H, NamedColor.blue)27            self.draw_symbol(1,2, NamedSymbol.E, NamedColor.yellow)28            self.draw_symbol(1,3, NamedSymbol.L, NamedColor.black)29            self.draw_symbol(1,4, NamedSymbol.L, NamedColor.green)30            self.draw_symbol(1,5, NamedSymbol.O, NamedColor.red)31        else:32            self.draw_symbol(1,1, NamedSymbol.none, NamedColor.blue)33            self.draw_symbol(1,2, NamedSymbol.none, NamedColor.yellow)34            self.draw_symbol(1,3, NamedSymbol.none, NamedColor.black)35            self.draw_symbol(1,4, NamedSymbol.none, NamedColor.green)36            self.draw_symbol(1,5, NamedSymbol.none, NamedColor.red)37def	main():38	game = MinimalGame(3, BRIDGES_USERNAME,  BRIDGES_API_KEY,)39	# start the game40	game.start()41if __name__ == "__main__":...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.