mirror of https://github.com/Kozea/pygal.git
Florian Mounier
9 years ago
32 changed files with 475 additions and 231 deletions
@ -0,0 +1,109 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2015 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library is distributed in the hope that it will be useful, but WITHOUT |
||||
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
||||
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
""" |
||||
Formatters to use with `value_formatter` and `x_value_formatter` configs |
||||
|
||||
""" |
||||
from datetime import datetime, date, time |
||||
from math import floor, log |
||||
from pygal._compat import u, to_str |
||||
from pygal.util import float_format |
||||
|
||||
|
||||
class Formatter(object): |
||||
pass |
||||
|
||||
|
||||
class HumanReadable(Formatter): |
||||
"""Format a number to engineer scale""" |
||||
ORDERS = u("yzafpnµm kMGTPEZY") |
||||
|
||||
def __init__(self, none_char=u('∅')): |
||||
self.none_char = none_char |
||||
|
||||
def __call__(self, val): |
||||
if val is None: |
||||
return self.none_char |
||||
order = val and int(floor(log(abs(val)) / log(1000))) |
||||
orders = self.ORDERS.split(" ")[int(order > 0)] |
||||
if order == 0 or order > len(orders): |
||||
return float_format(val / (1000 ** int(order))) |
||||
return ( |
||||
float_format(val / (1000 ** int(order))) + |
||||
orders[int(order) - int(order > 0)]) |
||||
|
||||
|
||||
class Significant(Formatter): |
||||
"""Show precision significant digit of float""" |
||||
def __init__(self, precision=10): |
||||
self.format = '%%.%dg' % precision |
||||
|
||||
def __call__(self, val): |
||||
if val is None: |
||||
return '' |
||||
return self.format % val |
||||
|
||||
|
||||
class Integer(Formatter): |
||||
"""Cast number to integer""" |
||||
|
||||
def __call__(self, val): |
||||
if val is None: |
||||
return '' |
||||
return '%d' % val |
||||
|
||||
|
||||
class Raw(Formatter): |
||||
"""Cast everything to string""" |
||||
|
||||
def __call__(self, val): |
||||
if val is None: |
||||
return '' |
||||
return to_str(val) |
||||
|
||||
|
||||
class IsoDateTime(Formatter): |
||||
"""Iso format datetimes""" |
||||
|
||||
def __call__(self, val): |
||||
if val is None: |
||||
return '' |
||||
return val.isoformat() |
||||
|
||||
|
||||
class Default(Significant, IsoDateTime, Raw): |
||||
"""Try to guess best format from type""" |
||||
|
||||
def __call__(self, val): |
||||
if val is None: |
||||
return '' |
||||
if isinstance(val, (int, float)): |
||||
return Significant.__call__(self, val) |
||||
if isinstance(val, (date, time, datetime)): |
||||
return IsoDateTime.__call__(self, val) |
||||
return Raw.__call__(self, val) |
||||
|
||||
# Formatters with default options |
||||
human_readable = HumanReadable() |
||||
significant = Significant() |
||||
integer = Integer() |
||||
raw = Raw() |
||||
|
||||
# Default config formatter |
||||
default = Default() |
||||
@ -0,0 +1,88 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2015 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library is distributed in the hope that it will be useful, but WITHOUT |
||||
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
||||
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
|
||||
"""Test formatters""" |
||||
|
||||
from pygal import formatters |
||||
from pygal._compat import u |
||||
|
||||
|
||||
def test_human_readable(): |
||||
"""Test human_readable formatter""" |
||||
f = formatters.human_readable |
||||
|
||||
assert f(1) == '1' |
||||
assert f(1.) == '1' |
||||
assert f(10) == '10' |
||||
assert f(12.5) == '12.5' |
||||
assert f(1000) == '1k' |
||||
assert f(5000) == '5k' |
||||
assert f(100000) == '100k' |
||||
assert f(1253) == '1.253k' |
||||
assert f(1250) == '1.25k' |
||||
|
||||
assert f(0.1) == '100m' |
||||
assert f(0.01) == '10m' |
||||
assert f(0.001) == '1m' |
||||
assert f(0.002) == '2m' |
||||
assert f(0.0025) == '2.5m' |
||||
assert f(0.0001) == u('100µ') |
||||
assert f(0.000123) == u('123µ') |
||||
assert f(0.00001) == u('10µ') |
||||
assert f(0.000001) == u('1µ') |
||||
assert f(0.0000001) == u('100n') |
||||
assert f(0.0000000001) == u('100p') |
||||
|
||||
assert f(0) == '0' |
||||
assert f(0.) == '0' |
||||
assert f(-1337) == '-1.337k' |
||||
assert f(-.000000042) == '-42n' |
||||
|
||||
|
||||
def test_human_readable_custom(): |
||||
"""Test human_readable formatter option""" |
||||
f = formatters.HumanReadable() |
||||
assert f(None) == '∅' |
||||
f = formatters.HumanReadable(none_char='/') |
||||
assert f(None) == '/' |
||||
|
||||
|
||||
def test_significant(): |
||||
"""Test significant formatter""" |
||||
f = formatters.significant |
||||
assert f(1) == '1' |
||||
assert f(1.) == '1' |
||||
assert f(-1.) == '-1' |
||||
assert f(10) == '10' |
||||
assert f(10000000000) == '1e+10' |
||||
assert f(100000000000) == '1e+11' |
||||
assert f(120000000000) == '1.2e+11' |
||||
|
||||
assert f(.1) == '0.1' |
||||
assert f(.01) == '0.01' |
||||
assert f(.0000000001) == '1e-10' |
||||
assert f(-.0000000001) == '-1e-10' |
||||
assert f(.0000000001002) == '1.002e-10' |
||||
|
||||
assert f(.0000000001002) == '1.002e-10' |
||||
|
||||
assert f(.12345678912345) == '0.1234567891' |
||||
assert f(.012345678912345) == '0.01234567891' |
||||
|
||||
assert f(12345678912345) == '1.234567891e+13' |
||||
Loading…
Reference in new issue