#!/usr/bin/env python3fromunittestimport TestCase, main
fromsympyimport symbols, plot
print('6.')
x = symbols('x')
m = symbols('m', real=True)
defdiscriminant(a, b, c):
return b **2-4* a * c
classMyTestCase(TestCase):
deftest_a(self):
self.assertTrue(discriminant(3, -1* m, -1).is_positive)
deftest_b(self):
d = discriminant(1, -6* m, 10* m **2)
self.assertTrue(d.subs({m: 0}).is_zero)
n = symbols('n', real=True, nonzero=True)
self.assertTrue(d.subs({m: n}).is_negative)
deftest_c(self):
d = discriminant(m, 4* m, 4* m +1)
self.assertTrue(
d.subs({m: symbols('n', negative=True)}).simplify().is_positive)
self.assertTrue(
d.subs({m: symbols('n', positive=True)}).simplify().is_negative)
p = plot(*[(m * x **2+4* m * x + (4* m +1)).subs({m: n}) for n in [-1, 1]],
(x, -5, 5),
ylim=(-5, 5),
legend=True,
show=False)
colors = ['red', 'green', 'blue', 'brown', 'orange',
'purple', 'pink', 'gray', 'skyblue', 'yellow']
for o, color inzip(p, colors):
o.line_color = color
p.show()
p.save(f'sample6.png')
if__name__=="__main__":
main()
入出力結果(Zsh、PowerShell、Terminal、Jupyter(IPython))
% ./sample6.py -v
6.
test_a (__main__.MyTestCase) ... ok
test_b (__main__.MyTestCase) ... ok
test_c (__main__.MyTestCase) ... ok
----------------------------------------------------------------------
Ran 3 tests in 0.310s
OK
%
0 コメント:
コメントを投稿