def setUp(self):
# color schemes with all different names
self.burg = styling.burg(bins=4)
self.burgYl = styling.burgYl(bins=4)
self.redOr = styling.redOr(bins=4)
self.orYel = styling.orYel(bins=4)
self.peach = styling.peach(bins=4)
self.pinkYl = styling.pinkYl(bins=4)
self.mint = styling.mint(bins=4)
self.bluGrn = styling.bluGrn(bins=4)
self.darkMint = styling.darkMint(bins=4)
self.emrld = styling.emrld(bins=4)
self.bluYl = styling.bluYl(bins=4)
self.teal = styling.teal(bins=4)
self.tealGrn = styling.tealGrn(bins=4)
self.purp = styling.purp(bins=4)
self.purpOr = styling.purpOr(bins=4)
self.sunset = styling.sunset(bins=4)
self.magenta = styling.magenta(bins=4)
self.sunsetDark = styling.sunsetDark(bins=4)
self.brwnYl = styling.brwnYl(bins=4)
self.armyRose = styling.armyRose(bins=4)
self.fall = styling.fall(bins=4)
self.geyser = styling.geyser(bins=4)
self.temps = styling.temps(bins=4)
self.tealRose = styling.tealRose(bins=4)
self.tropic = styling.tropic(bins=4)
self.earth = styling.earth(bins=4)
self.antique = styling.antique(bins=4)
self.bold = styling.bold(bins=4)
self.pastel = styling.pastel(bins=4)
self.prism = styling.prism(bins=4)
self.safe = styling.safe(bins=4)
self.vivid = styling.vivid(bins=4)
def test_get_scheme_cartocss(self):
"""styling.get_scheme_cartocss"""
# test on category
self.assertEqual(get_scheme_cartocss('acadia', self.vivid),
'ramp([acadia], cartocolor(Vivid), category(4), =)')
# test on quantative
self.assertEqual(get_scheme_cartocss('acadia', self.purp),
'ramp([acadia], cartocolor(Purp), quantiles(4), >)')
# test on custom
self.assertEqual(
get_scheme_cartocss(
'acadia',
styling.custom(('#FFF', '#888', '#000'),
bins=3,
bin_method='equal')),
'ramp([acadia], (#FFF,#888,#000), equal(3), >)')
# test with non-int quantification
self.assertEqual(
get_scheme_cartocss('acadia', styling.sunset([1, 2, 3])),
'ramp([acadia], cartocolor(Sunset), (1,2,3), >=)')
def test_line_styling(self): # pylint: disable=too-many-statements
"""layer.QueryLayer line styling"""
linelayer = QueryLayer('select * from lines', size=5)
linelayer.geom_type = 'line'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue('line-width: 5' in linelayer.cartocss)
size = 'size_col'
color = 'mag'
linelayer = QueryLayer('select * from lines', size=size, color=color)
linelayer.geom_type = 'line'
linelayer.style_cols['mag'] = 'number'
linelayer.style_cols['size_col'] = 'number'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue(
'line-width: ramp([size_col], range(1,5), quantiles(5))' in
linelayer.cartocss)
self.assertTrue(
'line-color: ramp([mag], cartocolor(Mint), quantiles(5), >)' in
linelayer.cartocss)
size = {'column': 'size_col'}
color = 'mag'
linelayer = QueryLayer('select * from lines', size=size, color=color)
linelayer.geom_type = 'line'
linelayer.style_cols['mag'] = 'number'
linelayer.style_cols['size_col'] = 'number'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue(
'line-width: ramp([size_col], range(1,5), quantiles(5))' in
linelayer.cartocss)
self.assertTrue(
'line-color: ramp([mag], cartocolor(Mint), quantiles(5), >)' in
linelayer.cartocss)
size = {'column': 'size_col', 'range': (5, 10)}
color = 'mag'
linelayer = QueryLayer('select * from lines', size=size, color=color)
linelayer.geom_type = 'line'
linelayer.style_cols['mag'] = 'number'
linelayer.style_cols['size_col'] = 'number'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue(
'line-width: ramp([size_col], range(5,10), quantiles(5))' in
linelayer.cartocss)
self.assertTrue(
'line-color: ramp([mag], cartocolor(Mint), quantiles(5), >)' in
linelayer.cartocss)
size = 1.5
color = 'mag'
linelayer = QueryLayer('select * from lines', size=size, color=color)
linelayer.geom_type = 'line'
linelayer.style_cols['mag'] = 'number'
linelayer.style_cols['size_col'] = 'number'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue('line-width: 1.5' in linelayer.cartocss)
self.assertTrue(
'line-color: ramp([mag], cartocolor(Mint), quantiles(5), >)' in
linelayer.cartocss)
size = {'column': 'size_col', 'range': [2, 6]}
color = {'column': 'mag', 'scheme': styling.sunset(7)}
linelayer = QueryLayer('select * from lines', size=size, color=color)
linelayer.geom_type = 'line'
linelayer.style_cols['mag'] = 'number'
linelayer.style_cols['size_col'] = 'number'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue(
'line-width: ramp([size_col], range(2,6), quantiles(5))' in
linelayer.cartocss)
self.assertTrue(
'line-color: ramp([mag], cartocolor(Sunset), quantiles(7), >)' in
linelayer.cartocss)
# size and color
size = {
'column': 'size_col',
'range': [2, 6],
'bin_method': BinMethod.jenks
}
color = {'column': 'mag', 'scheme': styling.sunset(7)}
linelayer = QueryLayer('select * from lines', size=size, color=color)
linelayer.geom_type = 'line'
linelayer.style_cols['mag'] = 'number'
linelayer.style_cols['size_col'] = 'number'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue('line-width: ramp([size_col], range(2,6), jenks(5))' in
linelayer.cartocss)
self.assertTrue(
'line-color: ramp([mag], cartocolor(Sunset), quantiles(7), >)' in
linelayer.cartocss)
# category lines
linelayer = QueryLayer('select * from lines',
color={
'column': 'mag',
'scheme': styling.antique(7)
})
linelayer.geom_type = 'line'
linelayer._setup([BaseMap(), linelayer], 1) # pylint: disable=protected-access
self.assertTrue(
'line-color: ramp([mag], cartocolor(Antique), category(7), =)' in
linelayer.cartocss)