def test_enable_context_with_arg(self):
ureg = UnitRegistry()
add_arg_ctxs(ureg)
q = 500 * ureg.meter
s = (ureg.speed_of_light / q).to("Hz")
self.assertRaises(ValueError, q.to, "Hz")
ureg.enable_contexts("lc", n=1)
self.assertEqual(q.to("Hz"), s)
ureg.enable_contexts("ab")
self.assertEqual(q.to("Hz"), s)
self.assertEqual(q.to("Hz"), s)
ureg.disable_contexts(1)
ureg.disable_contexts(1)
ureg.enable_contexts("ab")
self.assertRaises(ValueError, q.to, "Hz")
ureg.enable_contexts("lc", n=1)
self.assertEqual(q.to("Hz"), s)
ureg.disable_contexts(1)
self.assertRaises(ValueError, q.to, "Hz")
ureg.disable_contexts(1)
ureg.enable_contexts("lc")
self.assertRaises(TypeError, q.to, "Hz")
ureg.disable_contexts(1)
def test_known_context_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
ureg.enable_contexts("lc")
self.assertTrue(ureg._active_ctx)
self.assertTrue(ureg._active_ctx.graph)
ureg.disable_contexts(1)
self.assertFalse(ureg._active_ctx)
self.assertFalse(ureg._active_ctx.graph)
ureg.enable_contexts("lc", n=1)
self.assertTrue(ureg._active_ctx)
self.assertTrue(ureg._active_ctx.graph)
ureg.disable_contexts(1)
self.assertFalse(ureg._active_ctx)
self.assertFalse(ureg._active_ctx.graph)
def test_known_context_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
ureg.enable_contexts('lc')
self.assertTrue(ureg._active_ctx)
self.assertTrue(ureg._active_ctx.graph)
ureg.disable_contexts(1)
self.assertFalse(ureg._active_ctx)
self.assertFalse(ureg._active_ctx.graph)
ureg.enable_contexts('lc', n=1)
self.assertTrue(ureg._active_ctx)
self.assertTrue(ureg._active_ctx.graph)
ureg.disable_contexts(1)
self.assertFalse(ureg._active_ctx)
self.assertFalse(ureg._active_ctx.graph)
def test_known_context_enable(self, func_registry):
ureg = UnitRegistry()
add_ctxs(ureg)
ureg.enable_contexts("lc")
assert ureg._active_ctx
assert ureg._active_ctx.graph
ureg.disable_contexts(1)
assert not ureg._active_ctx
assert not ureg._active_ctx.graph
ureg.enable_contexts("lc", n=1)
assert ureg._active_ctx
assert ureg._active_ctx.graph
ureg.disable_contexts(1)
assert not ureg._active_ctx
assert not ureg._active_ctx.graph
def test_redefine(self):
ureg = UnitRegistry("""
foo = [d] = f = foo_alias
bar = 2 foo = b = bar_alias
baz = 3 bar = _ = baz_alias
asd = 4 baz
@context c
# Note how we're redefining a symbol, not the base name, as a
# function of another name
b = 5 f
""".splitlines())
# Units that are somehow directly or indirectly defined as a function of the
# overridden unit are also affected
foo = ureg.Quantity(1, "foo")
bar = ureg.Quantity(1, "bar")
asd = ureg.Quantity(1, "asd")
# Test without context before and after, to verify that the cache and units have
# not been polluted
for enable_ctx in (False, True, False):
with self.subTest(enable_ctx):
if enable_ctx:
ureg.enable_contexts("c")
k = 5
else:
k = 2
self.assertEqual(foo.to("b").magnitude, 1 / k)
self.assertEqual(foo.to("bar").magnitude, 1 / k)
self.assertEqual(foo.to("bar_alias").magnitude, 1 / k)
self.assertEqual(foo.to("baz").magnitude, 1 / k / 3)
self.assertEqual(bar.to("foo").magnitude, k)
self.assertEqual(bar.to("baz").magnitude, 1 / 3)
self.assertEqual(asd.to("foo").magnitude, 4 * 3 * k)
self.assertEqual(asd.to("bar").magnitude, 4 * 3)
self.assertEqual(asd.to("baz").magnitude, 4)
ureg.disable_contexts()
def test_graph_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
l = UnitsContainer({'[length]': 1.})
t = UnitsContainer({'[time]': -1.})
c = UnitsContainer({'[current]': 1.})
g_sp = defaultdict(set)
g_sp.update({l: set((t, )), t: set((l, ))})
g_ab = defaultdict(set)
g_ab.update({l: set((c, )), c: set((l, ))})
g = defaultdict(set)
g.update({l: set((t, c)), t: set((l, )), c: set((l, ))})
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('lc', n=1)
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g_ab)
ureg.disable_contexts(1)
ureg.enable_contexts('lc')
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab')
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('lc', 'ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab', 'lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
def test_graph_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
l = _freeze({"[length]": 1.0})
t = _freeze({"[time]": -1.0})
c = _freeze({"[current]": -1.0})
g_sp = defaultdict(set)
g_sp.update({l: set((t,)), t: set((l,))})
g_ab = defaultdict(set)
g_ab.update({l: set((c,)), c: set((l,))})
g = defaultdict(set)
g.update({l: set((t, c)), t: set((l,)), c: set((l,))})
ureg.enable_contexts("lc")
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts("lc", n=1)
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts("ab")
self.assertEqual(ureg._active_ctx.graph, g_ab)
ureg.disable_contexts(1)
ureg.enable_contexts("lc")
ureg.enable_contexts("ab")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts("ab")
ureg.enable_contexts("lc")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts("lc", "ab")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts("ab", "lc")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
def test_graph_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
l = _freeze({'[length]': 1.})
t = _freeze({'[time]': -1.})
c = _freeze({'[current]': -1.})
g_sp = defaultdict(set)
g_sp.update({
l: {
t,
},
t: {
l,
}
})
g_ab = defaultdict(set)
g_ab.update({
l: {
c,
},
c: {
l,
}
})
g = defaultdict(set)
g.update({
l: {t, c},
t: {
l,
},
c: {
l,
}
})
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('lc', n=1)
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g_ab)
ureg.disable_contexts(1)
ureg.enable_contexts('lc')
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab')
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('lc', 'ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab', 'lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
def test_graph_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
l = _freeze({'[length]': 1.})
t = _freeze({'[time]': -1.})
c = _freeze({'[current]': -1.})
g_sp = defaultdict(set)
g_sp.update({l: {t, },
t: {l, }})
g_ab = defaultdict(set)
g_ab.update({l: {c, },
c: {l, }})
g = defaultdict(set)
g.update({l: {t, c},
t: {l, },
c: {l, }})
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('lc', n=1)
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g_ab)
ureg.disable_contexts(1)
ureg.enable_contexts('lc')
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab')
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('lc', 'ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab', 'lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
def test_graph_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
l = UnitsContainer({"[length]": 1.0}) # noqa: E741
t = UnitsContainer({"[time]": -1.0})
c = UnitsContainer({"[current]": 1.0})
g_sp = defaultdict(set)
g_sp.update({l: {t}, t: {l}})
g_ab = defaultdict(set)
g_ab.update({l: {c}, c: {l}})
g = defaultdict(set)
g.update({l: {t, c}, t: {l}, c: {l}})
ureg.enable_contexts("lc")
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts("lc", n=1)
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts("ab")
self.assertEqual(ureg._active_ctx.graph, g_ab)
ureg.disable_contexts(1)
ureg.enable_contexts("lc")
ureg.enable_contexts("ab")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts("ab")
ureg.enable_contexts("lc")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts("lc", "ab")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts("ab", "lc")
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
def test_graph_enable(self):
ureg = UnitRegistry()
add_ctxs(ureg)
l = UnitsContainer({'[length]': 1.})
t = UnitsContainer({'[time]': -1.})
c = UnitsContainer({'[current]': 1.})
g_sp = defaultdict(set)
g_sp.update({l: set((t,)),
t: set((l,))})
g_ab = defaultdict(set)
g_ab.update({l: set((c,)),
c: set((l,))})
g = defaultdict(set)
g.update({l: set((t, c)),
t: set((l,)),
c: set((l,))})
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('lc', n=1)
self.assertEqual(ureg._active_ctx.graph, g_sp)
ureg.disable_contexts(1)
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g_ab)
ureg.disable_contexts(1)
ureg.enable_contexts('lc')
ureg.enable_contexts('ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab')
ureg.enable_contexts('lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('lc', 'ab')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
ureg.enable_contexts('ab', 'lc')
self.assertEqual(ureg._active_ctx.graph, g)
ureg.disable_contexts(2)
