def test_about(capfd):
"""sf.about works."""
sf.about()
out, err = capfd.readouterr()
# substantial output (actual length varies)
assert len(err) == 0
assert len(out) > 300
assert "Strawberry Fields version" in out
sf_version_match = re.search(r"Strawberry Fields version:\s+([\S]+)\n",
out).group(1)
assert sf_version_match == sf.version()
assert "Numpy version" in out
assert "Scipy version" in out
assert "Hafnian version" in out
assert "Blackbird version" in out
def test_power_positive_frac(self):
a = np.random.random()
x = np.random.random()
q = self.eng.register
rrt = power(q[0], a)
self.assertEqual(type(rrt), sf.engine.RegRefTransform)
self.assertEqual(rrt.func(x), x**a)
def test_power_negative_int(self):
a = -3
x = np.random.random()
q = self.eng.register
rrt = power(q[0], a)
self.assertEqual(type(rrt), sf.engine.RegRefTransform)
self.assertEqual(rrt.func(x), x**a)
if __name__ == '__main__':
print('Testing Strawberry Fields version ' + sf.version() + ', Utils.')
# run the tests in this file
suite = unittest.TestSuite()
tests = [InitialStates, FockInitialStates, ConvertFunctions]
for t in tests:
ttt = unittest.TestLoader().loadTestsFromTestCase(t)
suite.addTests(ttt)
unittest.TextTestRunner().run(suite)
def test_optimization(self):
filename = os.path.join(examples_folder,"optimization.py")
soln = b"Value at step 49: 0.36787864565849304\n"
# tests whether the output fock probability are as
# expected, as written in soln
p = Popen(["python3", filename], stdout=PIPE,
preexec_fn=os.setsid)
out = p.stdout.readlines()[-1]
self.assertEqual(out, soln)
p.terminate()
if __name__ == '__main__':
print('Testing Strawberry Fields version ' + sf.version() + ', examples.')
# run the tests in this file
suite = unittest.TestSuite()
tests = [
GaussianExamples,
FockExamples,
TFExamples
]
for t in tests:
ttt = unittest.TestLoader().loadTestsFromTestCase(t)
suite.addTests(ttt)
unittest.TextTestRunner().run(suite)