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)