def read_luncher_prefs(filename):
name = os.path.basename(filename).replace(".prefs", "")
prefs = LuncherPreferences(name)
with open(filename) as f:
for line in f:
line = line.strip()
if line.startswith('#'):
continue
weight, name = line.split(',')
prefs.update(name, weight)
prefs.normalize()
sys.stderr.write("Loaded %s\n" % filename)
return prefs
class TestLuncherPreferences(unittest.TestCase):
def setUp(self):
self.luncherprefs = LuncherPreferences("ryan")
def test_add_over_100(self):
self.luncherprefs.update("pizza", 50)
self.luncherprefs.update("cafe", 60)
self.luncherprefs.normalize()
self.assertEqual(sum(self.luncherprefs.preferences.values()), 100)
self.assertEqual(round(self.luncherprefs.preferences["pizza"]), 45)
self.assertEqual(round(self.luncherprefs.preferences["cafe"]), 55)
def setUp(self):
self.luncherprefs = LuncherPreferences("ryan")
