def test_string(self): df = DataFrame({'fruit': ['apple', 'banana']} ) table = dataframe_to_gviz(df) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("string", "fruit", "fruit"); data.addRows(2); data.setCell(0, 0, "apple"); data.setCell(1, 0, "banana"); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_boolean(self): df = DataFrame({'booleans': [True, False]} ) table = dataframe_to_gviz(df) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("boolean", "booleans", "booleans"); data.addRows(2); data.setCell(0, 0, true); data.setCell(1, 0, false); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_float(self): df = DataFrame({'floats': [5, 3.456]} ) table = dataframe_to_gviz(df) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("number", "floats", "floats"); data.addRows(2); data.setCell(0, 0, 5.0); data.setCell(1, 0, 3.456); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_null_006(self): """ test None """ df = DataFrame({'fruit': ['apple', None, 'plum']}) table = dataframe_to_gviz(df, allow_nulls=True) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("string", "fruit", "fruit"); data.addRows(3); data.setCell(0, 0, "apple"); data.setCell(2, 0, "plum"); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_null_004(self): """ test NaT """ df = DataFrame({'dates': [datetime(2015, 03, 21), pd.NaT, datetime(1999, 12, 01 )]} ) table = dataframe_to_gviz(df, allow_nulls=True) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("date", "dates", "dates"); data.addRows(3); data.setCell(0, 0, new Date(2015,2,21)); data.setCell(2, 0, new Date(1999,11,1)); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_null_002(self): """ test nan """ df = DataFrame({'nums': [3,np.nan, 5]}) table = dataframe_to_gviz(df, allow_nulls=True) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("number", "nums", "nums"); data.addRows(3); data.setCell(0, 0, 3.0); data.setCell(2, 0, 5.0); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_date_002(self): # datetime with y, m, d, h, m, s df = DataFrame({'dates': [datetime(2015, 03, 21, 15, 59, 45), datetime(1999, 12, 01, 4, 15, 21)]} ) table = dataframe_to_gviz(df) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("date", "dates", "dates"); data.addRows(2); data.setCell(0, 0, new Date(2015,2,21)); data.setCell(1, 0, new Date(1999,11,1)); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_date_004(self): # pandas Timestamp dates df = DataFrame({'dates': [pd.to_datetime('2015-10-04'), pd.to_datetime('1999-03-21')]} ) table = dataframe_to_gviz(df) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("date", "dates", "dates"); data.addRows(2); data.setCell(0, 0, new Date(2015,9,4)); data.setCell(1, 0, new Date(1999,2,21)); """ self.assertEqual(table.ToJSCode("data"), expected)
def test_date_005(self): # date() with y, m, d ##### NOTE: datetime.date() IS TREATED AS A STRING IN PANDAS ###### df = DataFrame({'dates': [date(2015, 03, 21), date(1999, 12, 01 )]} ) table = dataframe_to_gviz(df) # print table.ToJSCode('data') expected = """var data = new google.visualization.DataTable(); data.addColumn("string", "dates", "dates"); data.addRows(2); data.setCell(0, 0, "2015-03-21"); data.setCell(1, 0, "1999-12-01"); """ self.assertEqual(table.ToJSCode("data"), expected)