From e49146752ff322c53430419749be53b741f02540 Mon Sep 17 00:00:00 2001
From: Marco Pesenti Gritti <marco@localhost.localdomain>
Date: Thu, 26 Apr 2007 09:31:41 +0000
Subject: More work on the new graphics API.

---
(limited to 'tests')

diff --git a/tests/test-ui.py b/tests/test-ui.py
index 26f3d12..38013fb 100755
--- a/tests/test-ui.py
+++ b/tests/test-ui.py
@@ -18,11 +18,9 @@
 
 import gtk
 
-from sugar.graphics2.window import Window
-
-class ActivityToolbar(gtk.Toolbar):
-    def __init__(self):
-        gtk.Toolbar.__init__(self)
+from sugar.graphics.window import Window
+from sugar.graphics.toolbutton import ToolButton
+from sugar.graphics.toolbox import ActivityToolbar
 
 class EditToolbar(gtk.Toolbar):
     def __init__(self):
@@ -32,8 +30,7 @@ class TextToolbar(gtk.Toolbar):
     def __init__(self):
         gtk.Toolbar.__init__(self)
 
-        button = gtk.ToolButton()
-        button.set_icon_name('text-format-bold')
+        button = ToolButton('text-format-bold')
         self.insert(button, -1)
         button.show()
 
--
cgit v0.9.1