Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
    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)
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
    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)
Exemplo n.º 5
0
    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)
Exemplo n.º 6
0
    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)
Exemplo n.º 7
0
    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)
Exemplo n.º 8
0
    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)
Exemplo n.º 9
0
    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)