def test1(self): print("Testing get/set of printing strategy") self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy) set_printing_strategy("no_unit") self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) set_printing_strategy("default") self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy) set_printing_strategy("no_units") self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) set_printing_strategy("with_units") self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy) set_printing_strategy("formal") self.assertEqual(get_current_printing_strategy(), console.FormalPrintingStrategy) set_printing_strategy("nbody") self.assertEqual(get_current_printing_strategy(), console.NBodyPrintingStrategy) set_printing_strategy(console.NoUnitsPrintingStrategy) self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) set_printing_strategy(console.DefaultPrintingStrategy) self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy)
def test1(self): print "Testing get/set of printing strategy" self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy) set_printing_strategy("no_unit") self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) set_printing_strategy("default") self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy) set_printing_strategy("no_units") self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) set_printing_strategy("with_units") self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy) set_printing_strategy("formal") self.assertEqual(get_current_printing_strategy(), console.FormalPrintingStrategy) set_printing_strategy("nbody") self.assertEqual(get_current_printing_strategy(), console.NBodyPrintingStrategy) set_printing_strategy(console.NoUnitsPrintingStrategy) self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) set_printing_strategy(console.DefaultPrintingStrategy) self.assertEqual(get_current_printing_strategy(), console.DefaultPrintingStrategy)
def test2(self): print "Testing no units printing strategy with SI quantities" mass = 1.0 | units.kg acc = 9.8 | units.m / units.s**2 position = [1, 2.0, 3] | units.m mass_in_g = mass.as_quantity_in(units.g) pi = 3.14 | units.none self.assertEqual(str(mass), "1.0 kg") self.assertEqual(str(acc), "9.8 m / (s**2)") self.assertEqual(str(position), "[1.0, 2.0, 3.0] m") self.assertEqual(str(mass_in_g), "1000.0 g") self.assertEqual(str(pi), "3.14 none") set_printing_strategy("no_unit") self.assertEqual(get_current_printing_strategy(), console.NoUnitsPrintingStrategy) self.assertEqual(str(mass), "1.0") self.assertEqual(str(acc), "9.8") self.assertEqual(str(position), "[1.0, 2.0, 3.0]") self.assertEqual(str(mass_in_g), "1000.0") self.assertEqual(str(pi), "3.14") set_printing_strategy("default")