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")