Exemplo n.º 1
0
	def setUp(self):
		self.options = dict()
		self.options['title'] = "A Title"
		self.options['subtitle'] = "A Subtitle"
		self.options['theme'] = Theme()
		self.options['stylesheet'] = StyleSheet()
		self.options['xaxis_title'] = 'XAxis Title'
		self.options['yaxis_title'] = 'YAxis Title'
		# declare some mock tiles.
		self.t1 = Tile({'value': 105, 'alpha': 0.45, 'rgb': '#f00000'})
		self.t2 = Tile({'value': 10, 'alpha': 0.2566, 'rgb': '#f10000'})
		self.t3 = Tile({'value': 123, 'alpha': 0.89, 'rgb': '#f20000'})
		self.t4 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t5 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t6 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t7 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t8 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t9 = Tile({'value': 100, 'alpha': 0.11, 'rgb': '#f80000'})
		self.options['data'] = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		self.t1, 	self.t2, 	self.t3],
			['Midday', 		self.t4, 	self.t5, 	self.t6],
			['Afternoon',	self.t7, 	self.t8, 	self.t9],
		]
		self.heatmap = HeatMap(self.options)
Exemplo n.º 2
0
class TestGetTile(unittest.TestCase):
	def setUp(self):
		self.options = dict()
		self.options['title'] = "A Title"
		self.options['subtitle'] = "A Subtitle"
		self.options['theme'] = Theme()
		self.options['stylesheet'] = StyleSheet()
		self.options['xaxis_title'] = 'XAxis Title'
		self.options['yaxis_title'] = 'YAxis Title'
		# declare some mock tiles.
		self.t1 = Tile({'value': 105, 'alpha': 0.45, 'rgb': '#f00000'})
		self.t2 = Tile({'value': 10, 'alpha': 0.2566, 'rgb': '#f10000'})
		self.t3 = Tile({'value': 123, 'alpha': 0.89, 'rgb': '#f20000'})
		self.t4 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t5 = Tile({'value': 35, 'alpha': 0.97, 'rgb': '#f40000'})
		self.t6 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t7 = Tile({'value': 67.43, 'alpha': 0.522, 'rgb': '#f60000'})
		self.t8 = Tile({'value': 49, 'alpha': 0.567, 'rgb': '#f70000'})
		self.t9 = Tile({'value': 100, 'alpha': 0.11, 'rgb': '#f80000'})
		self.options['data'] = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		self.t1, 	self.t2, 	self.t3],
			['Midday', 		self.t4, 	self.t5, 	self.t6],
			['Afternoon',	self.t7, 	self.t8, 	self.t9],
		]
		self.heatmap = HeatMap(self.options)


	def test_get_tile_A(self):
		self.assertEqual(
			self.heatmap.get_tile(0,0),
			'')
		self.assertEqual(
			self.heatmap.get_tile(1,0),
			'Morning')
		self.assertEqual(
			self.heatmap.get_tile(2,0),
			'Midday')
		self.assertEqual(
			self.heatmap.get_tile(2,1),
			self.t4)
		self.assertEqual(
			self.heatmap.get_tile(3,2),
			self.t8)



	def test_get_tile_B(self):
		self.assertRaises(
			IndexError,
			self.heatmap.get_tile,
			5,0)
		self.assertRaises(
			IndexError,
			self.heatmap.get_tile,
			100,0)
		self.assertRaises(
			IndexError,
			self.heatmap.get_tile,
			0,4)
Exemplo n.º 3
0
	def test_heatmap_setter_A(self):
		mock_data1 = [
			["Month","Web Traffic","Call to actions"],
			["January",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["February",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["March",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["April",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["May",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["June",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["July",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["August",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["September",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["October",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["November",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["December",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})]
		]
		mock_data2 = [
			["Month","Web Traffic","Call to actions"],
			["January",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["February",Tile({'value': 60.67, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["March",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["April",Tile({'value': 20, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 10, 'rgb': '#a45aff', 'alpha': 0.56})],
			["May",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.345})],
			["June",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 15.89, 'rgb': '#a45aff', 'alpha': 0.34})],
			["July",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["August",Tile({'value': 67.89, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["September",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.553})],
			["October",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.45}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.11})],
			["November",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.5}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.911})],
			["December",Tile({'value': 908, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.734})]
			]
		options1 = {
			'data': mock_data1,
			'title': 'My First Heatmap title',
			'subtitle': 'The first heatmap subtitle',
			'theme': Theme(),
			'stylesheet': StyleSheet(),
			'xaxis_title': 'Title of X-Axis',
			'yaxis_title': 'Title of Y-Axis'
		}
		options2 = {
			'data': mock_data2,
			'title': 'My First Heatmap title',
			'subtitle': 'The first heatmap subtitle',
			'theme': Theme(),
			'stylesheet': StyleSheet(),
			'xaxis_title': 'Title of X-Axis',
			'yaxis_title': 'Title of Y-Axis'
		}
		heatmap1 = HeatMap(options1)
		heatmap2 = HeatMap(options2)
		file_output_writer=FileOutputWriter(filepath="filepath.file",heatmap=heatmap1)
		self.assertEqual(file_output_writer.heatmap,
			heatmap1)
		file_output_writer.heatmap = heatmap2
		self.assertEqual(file_output_writer.heatmap,
			heatmap2)
Exemplo n.º 4
0
	def test_file_path_setter_A(self):
		mock_data = [
			["Month","Web Traffic","Call to actions"],
			["January",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["February",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["March",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["April",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["May",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["June",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["July",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["August",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["September",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["October",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["November",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["December",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})]
		]
		options = {
			'data': mock_data,
			'title': 'My First Heatmap title',
			'subtitle': 'The first heatmap subtitle',
			'theme': Theme(),
			'stylesheet': StyleSheet(),
			'xaxis_title': 'Title of X-Axis',
			'yaxis_title': 'Title of Y-Axis'
		}
		heatmap = HeatMap(options)
		file_output_writer=FileOutputWriter(filepath="filepath.file",heatmap=heatmap)
		self.assertEqual(file_output_writer.filepath,
			"filepath.file")
		file_output_writer.filepath = "myfile.html"
		self.assertEqual(file_output_writer.filepath,
			"myfile.html")
Exemplo n.º 5
0
	def test_write_A(self):
		mock_data = [
			["Month","Web Traffic","Call to actions"],
			["January",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["February",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["March",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["April",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["May",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})]
		]
		stylesheet = StyleSheet()
		options = {
			'data': mock_data,
			'title': 'My First Heatmap title',
			'subtitle': 'The first heatmap subtitle',
			'theme': Theme(),
			'stylesheet': stylesheet,
			'xaxis_title': 'Title of X-Axis',
			'yaxis_title': 'Title of Y-Axis'
		}
		filepath = "myoutputfile.html"
		heatmap = HeatMap(options)
		html_writer = HTMLWriter(filepath=filepath, heatmap=heatmap)
		html_writer.write()
		self.assertTrue(os.path.exists(filepath))
		os.remove(filepath)
Exemplo n.º 6
0
	def test___init__A(self):
		mock_data = [
			["Month","Web Traffic","Call to actions"],
			["January",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["February",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["March",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["April",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["May",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),
				Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})]
		]
		stylesheet = StyleSheet()
		options = {
			'data': mock_data,
			'title': 'My First Heatmap title',
			'subtitle': 'The first heatmap subtitle',
			'theme': Theme(),
			'stylesheet': stylesheet,
			'xaxis_title': 'Title of X-Axis',
			'yaxis_title': 'Title of Y-Axis'
		}
		filepath = "myfilepath.html"
		heatmap = HeatMap(options)
		

		html_writer = HTMLWriter(filepath=filepath, heatmap=heatmap)

		self.assertRaises(TypeError, HTMLWriter, filepath=filepath,
			heatmap=list())
Exemplo n.º 7
0
	def test___init__A(self):
		options = dict()
		options['title'] = "A Title"
		options['subtitle'] = "A Subtitle"
		options['theme'] = Theme()
		options['stylesheet'] = StyleSheet()
		options['xaxis_title'] = 'XAxis Title'
		options['yaxis_title'] = 'YAxis Title'

		# declare some mock tiles.
		t1 = Tile({'value': 105, 'alpha': 0.45, 'rgb': '#f00000'})
		t2 = Tile({'value': 10, 'alpha': 0.2566, 'rgb': '#f10000'})
		t3 = Tile({'value': 123, 'alpha': 0.89, 'rgb': '#f20000'})
		t4 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		t5 = Tile({'value': 35, 'alpha': 0.97, 'rgb': '#f40000'})
		t6 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		t7 = Tile({'value': 67.43, 'alpha': 0.522, 'rgb': '#f60000'})
		t8 = Tile({'value': 49, 'alpha': 0.567, 'rgb': '#f70000'})
		t9 = Tile({'value': 100, 'alpha': 0.11, 'rgb': '#f80000'})

		options['data'] = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		t1, 		t2, 				t3],
			['Midday', 		t4, 		t5, 				t6],
			['Afternoon',	t7, 		t8, 				t9],
		]

		heatmap = HeatMap(options)
Exemplo n.º 8
0
class TestGetDistinctTiles(unittest.TestCase):
	def setUp(self):
		self.options = dict()
		self.options['title'] = "A Title"
		self.options['subtitle'] = "A Subtitle"
		self.options['theme'] = Theme()
		self.options['stylesheet'] = StyleSheet()
		self.options['xaxis_title'] = 'XAxis Title'
		self.options['yaxis_title'] = 'YAxis Title'
		# declare some mock tiles.
		self.t1 = Tile({'value': 105, 'alpha': 0.45, 'rgb': '#f00000'})
		self.t2 = Tile({'value': 10, 'alpha': 0.2566, 'rgb': '#f10000'})
		self.t3 = Tile({'value': 123, 'alpha': 0.89, 'rgb': '#f20000'})
		self.t4 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t5 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t6 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t7 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t8 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t9 = Tile({'value': 100, 'alpha': 0.11, 'rgb': '#f80000'})
		self.options['data'] = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		self.t1, 	self.t2, 	self.t3],
			['Midday', 		self.t4, 	self.t5, 	self.t6],
			['Afternoon',	self.t7, 	self.t8, 	self.t9],
		]
		self.heatmap = HeatMap(self.options)

	def test_get_distinct_tiles_A(self):
		distinct_tiles = self.heatmap.get_distinct_tiles()
		the_tiles = [self.t1.value, self.t2.value, self.t3.value,
			self.t4.value,self.t6.value, self.t9.value]
		self.assertEqual(distinct_tiles, the_tiles)
Exemplo n.º 9
0
 def generate_heatmap(self, data_grid):
     ''' creates an object of type HeatMap '''
     new_data = list()
     # get the maximum value out of the
     # entire DataGrid object passed.
     max_value = data_grid.get_max_value()
     min_value = data_grid.get_min_value()
     heatmap = list()
     grid = data_grid.grid
     # generate a heatmap Tile for every
     # data item detected as 'Data' type
     # from the passed DataGrid object
     for row in grid:
         final_row = list()
         for item in range(len(row)):
             if(type(row[item]) is Data):                    
                 new_tile = self.generate_tile(row[item],
                     min_value, max_value,
                     self.__theme)
                 final_row.append(new_tile)
             else:
                 final_row.append(row[item])
         heatmap.append(final_row)
     # create a HeatMap object passing options
     # and especially the DataGrid object.
     heatmap_obj = HeatMap({'data': heatmap,
         'title': self.__title,
         'subtitle': self.__subtitle,
         'theme': self.__theme,
         'stylesheet': self.__stylesheet,
         'xaxis_title': self.__xaxistitle,
         'yaxis_title': self.__yaxistitle,
         'rowcol_headers': data_grid.rowcolheaders,
         'xaxis_labels': self.__xaxislabels,
         'yaxis_labels': self.__yaxislabels,
         'data_formatter': self.__data_formatter,
         'grouping': self.__grouping})
     return heatmap_obj
Exemplo n.º 10
0
	def test_write_A(self):
		mock_data = [
			["Month","Web Traffic","Call to actions"],
			["January",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["February",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["March",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["April",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})],
			["May",Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56}),Tile({'value': 60, 'rgb': '#a45aff', 'alpha': 0.56})]
		]
		stylesheet = StyleSheet()
		options = {
			'data': mock_data,
			'title': 'My First Heatmap title',
			'subtitle': 'The first heatmap subtitle',
			'theme': Theme(),
			'stylesheet': stylesheet,
			'xaxis_title': 'Title of X-Axis',
			'yaxis_title': 'Title of Y-Axis'
		}
		heatmap = HeatMap(options)
		file_output_writer=FileOutputWriter(filepath="filepath.file",heatmap=heatmap)
		self.assertRaises(NotImplementedError,
			file_output_writer.write)
Exemplo n.º 11
0
class TestEqual(unittest.TestCase):
	def setUp(self):
		self.options1 = dict()
		self.options1['title'] = "A Title"
		self.options1['subtitle'] = "A Subtitle"
		self.options1['theme'] = Theme()
		self.options1['stylesheet'] = StyleSheet()
		self.options1['xaxis_title'] = 'XAxis Title'
		self.options1['yaxis_title'] = 'YAxis Title'

		self.options2 = dict()
		self.options2['title'] = "A Title"
		self.options2['subtitle'] = "A Subtitle"
		self.options2['theme'] = Theme()
		self.options2['stylesheet'] = StyleSheet()
		self.options2['xaxis_title'] = 'XAxis Title'
		self.options2['yaxis_title'] = 'YAxis Title'

		self.options3 = dict()
		self.options3['title'] = "A Title"
		self.options3['subtitle'] = "A Subtitle"
		self.options3['theme'] = Theme()
		self.options3['stylesheet'] = StyleSheet()
		self.options3['xaxis_title'] = 'XAxis Title'
		self.options3['yaxis_title'] = 'YAxis Title'


		# declare some mock tiles.
		self.t1 = Tile({'value': 105, 'alpha': 0.45, 'rgb': '#f00000'})
		self.t2 = Tile({'value': 10, 'alpha': 0.2566, 'rgb': '#f10000'})
		self.t3 = Tile({'value': 123, 'alpha': 0.89, 'rgb': '#f20000'})
		self.t4 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t5 = Tile({'value': 45, 'alpha': 0.15, 'rgb': '#f30000'})
		self.t6 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t7 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t8 = Tile({'value': 89, 'alpha': 0.24, 'rgb': '#f50000'})
		self.t9 = Tile({'value': 100, 'alpha': 0.11, 'rgb': '#f80000'})

		# create some data lists.
		self.data1 = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		self.t1, 	self.t2, 	self.t3],
			['Midday', 		self.t4, 	self.t5, 	self.t6],
			['Afternoon',	self.t7, 	self.t8, 	self.t9],
		]

		self.data2 = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		self.t4, 	self.t2, 	self.t3],
			['Midday', 		self.t1, 	self.t5, 	self.t6],
			['Afternoon',	self.t6, 	self.t8, 	self.t1],
		]


		self.data3 = [
			['',			'Monday',	'Tuesday',	'Wednesday'],
			['Morning',		self.t1, 	self.t2, 	self.t3],
			['Midday', 		self.t4, 	self.t5, 	self.t6],
			['Afternoon',	self.t7, 	self.t8, 	self.t9],
		]


		self.options1['data'] = self.data1
		self.options2['data'] = self.data2
		self.options3['data'] = self.data3

		self.heatmap1 = HeatMap(self.options1)
		self.heatmap2 = HeatMap(self.options2)
		self.heatmap3 = HeatMap(self.options3)

	def test___eq__A(self):
		self.assertFalse(self.heatmap1.__eq__(self.heatmap2))
		self.assertFalse(self.heatmap2.__eq__(self.heatmap3))
		self.assertTrue(self.heatmap1.__eq__(self.heatmap3))

	def test___eq__B(self):
		self.assertNotEqual(self.heatmap1, self.heatmap2)
		self.assertNotEqual(self.heatmap2, self.heatmap3)
		self.assertEqual(self.heatmap1, self.heatmap3)
Exemplo n.º 12
0
 def test___init__B(self):
     mock_data = [["Month", "Web Traffic", "Call to actions"],
                  [
                      "January",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "February",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "March",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "April",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "May",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "June",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "July",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "August",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "September",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "October",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "November",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ],
                  [
                      "December",
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      }),
                      Tile({
                          'value': 60,
                          'rgb': '#a45aff',
                          'alpha': 0.56
                      })
                  ]]
     options = {
         'data': mock_data,
         'title': 'My First Heatmap title',
         'subtitle': 'The first heatmap subtitle',
         'theme': Theme(),
         'stylesheet': StyleSheet(),
         'xaxis_title': 'Title of X-Axis',
         'yaxis_title': 'Title of Y-Axis'
     }
     heatmap = HeatMap(options)
     HeatMapWindow(self.parent, heatmap=heatmap)