def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
self.test_actions = [TakeMovieAction1(directory = directory,
length = 5,
name = filename),
TakeMovieAction2(directory = directory,
length = 2,
name = filename,
overwrite = False),
TakeMovieAction2(directory = directory,
length = 2,
name = filename,
overwrite = False,
test_mode = True),
TakeMovieAction1(directory = directory,
length = 5,
name = filename)]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
self.test_actions = [TakeMovieAction3(directory = directory,
length = 50,
name = filename,
parameters = "default",
test_mode = True)]
def test_hal_film_1():
# This is expected to record a movie called 'movie_01.dax'
halTest(config_xml = "none_classic_config.xml",
class_name = "FilmTest1",
test_module = "storm_control.test.hal.film_tests")
# Check that the movie is the right length.
movie = datareader.inferReader(os.path.join(test.dataDirectory(), "movie_01.dax"))
assert(movie.filmSize() == [512, 512, 10])
def test_hal_film():
# This is expected to record a movie called 'movie_01.dax'
halTest(config_xml="none_classic_config.xml",
class_name="FilmTest1",
test_module="storm_control.test.hal.film_tests")
# Check that the movie is the right length.
movie = datareader.inferReader(
os.path.join(test.dataDirectory(), "movie_01.dax"))
assert (movie.filmSize() == [512, 512, 10])
def test_hal_film_5():
"""
Test repeated film acquisition.
"""
halTest(config_xml = "none_classic_config.xml",
class_name = "FilmTest4",
test_module = "storm_control.test.hal.film_tests")
# Check that the final movie is correct.
movie = datareader.inferReader(os.path.join(test.dataDirectory(), "movie_04.dax"))
assert(movie.filmSize() == [512, 512, 1])
def test_v1_1():
"""
This just tests that the XML file is created without crashing. The positions
file is the expected format, but with a few blank lines at the end to make
it more interesting.
"""
input_xml = test.daveXmlFilePathAndName("v1_generator_test.xml")
input_positions = test.daveXmlFilePathAndName(
"v1_generator_test_positions.txt")
output_xml = os.path.join(test.dataDirectory(), "dave_sequence.xml")
v1Generator.generate(None, input_xml, input_positions, output_xml)
def __init__(self, **kwds):
super().__init__(**kwds)
self.test_actions = [SetProgressionAction1(filename = test.halXmlFilePathAndName("prog_test.power"),
prog_type = "file"),
testActions.Timer(500),
SetProgressionAction1(prog_type = "lockedout"),
testActions.Timer(500),
TakeMovieAction1(directory = test.dataDirectory(),
length = 5,
name = "movie_01"),
testActions.Timer(500)]
def test_hal_film_5():
"""
Test repeated film acquisition.
"""
halTest(config_xml="none_classic_config.xml",
class_name="FilmTest4",
test_module="storm_control.test.hal.film_tests")
# Check that the final movie is correct.
movie = datareader.inferReader(
os.path.join(test.dataDirectory(), "movie_04.dax"))
assert (movie.filmSize() == [512, 512, 1])
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
self.test_actions = [testActions.SetDirectory(directory = directory),
testActions.Record(filename = filename)]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
self.reps = 10
self.test_actions = [
testActions.RemoveFile(directory=directory, name=filename),
tcpTests.TakeMovieAction1(directory=directory,
length=5,
name=filename)
]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
p_name = "256x256"
self.test_actions = [testActions.LoadParameters(filename = test.halXmlFilePathAndName(p_name + ".xml")),
TakeMovieAction4(directory = directory,
length = 50,
name = filename,
parameters = p_name,
test_mode = True)]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filenames = ["movie_01", "movie_02"]
# Remove old movies (if any).
for elt in filenames:
fullname = os.path.join(directory, elt + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
self.test_actions = [testActions.SetDirectory(directory = directory),
testActions.LoadParameters(filename = test.halXmlFilePathAndName("feed_test.xml")),
testActions.SetParameters(p_name = 0),
testActions.Timer(timeout = 200),
testActions.Record(filename = filenames[0]),
testActions.Timer(timeout = 200),
testActions.LoadParameters(filename = os.path.join(test.dataDirectory(), filenames[0] + ".xml")),
testActions.SetParameters(p_name = 0),
testActions.Timer(timeout = 200),
testActions.Record(filename = filenames[1]),
testActions.Timer(timeout = 200)]
def test_hal_film_2():
# This is expected to record several movies with names starting with 'movie_02'
halTest(config_xml = "none_classic_config.xml",
class_name = "FilmTest2",
test_module = "storm_control.test.hal.film_tests")
# Check that the movies are the right length.
for name, size in [["movie_02.dax", [512, 512, 10]],
["movie_02_average.dax", [512, 512, 1]],
["movie_02_interval.dax", [508, 256, 2]],
["movie_02_slice1.dax", [64, 65, 10]]]:
movie = datareader.inferReader(os.path.join(test.dataDirectory(), name))
assert(movie.filmSize() == size)
def test_hal_film_2():
# This is expected to record several movies with names starting with 'movie_02'
halTest(config_xml="none_classic_config.xml",
class_name="FilmTest2",
test_module="storm_control.test.hal.film_tests")
# Check that the movies are the right length.
for name, size in [["movie_02.dax", [512, 512, 10]],
["movie_02_average.dax", [512, 512, 1]],
["movie_02_interval.dax", [508, 256, 2]],
["movie_02_slice1.dax", [64, 65, 10]]]:
movie = datareader.inferReader(os.path.join(test.dataDirectory(),
name))
assert (movie.filmSize() == size)
def test_hal_film_7():
# This is expected to record several movies with names starting
# with 'movie_01' and 'movie_02'.
halTest(config_xml = "none_classic_config.xml",
class_name = "FilmTest6",
test_module = "storm_control.test.hal.film_tests")
# Check that the movies are the right length.
for name, size in [["movie_01.dax", [256, 256, 10]],
["movie_01_slice1.dax", [128, 128, 10]],
["movie_02.dax", [256, 256, 10]],
["movie_02_slice1.dax", [128, 128, 10]]]:
movie = datareader.inferReader(os.path.join(test.dataDirectory(), name))
assert(movie.filmSize() == size)
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_04"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
os.remove(os.path.join(directory, filename + ".xml"))
self.test_actions = [testActions.SetDirectory(directory = directory)]
for i in range(10):
self.test_actions.append(testActions.Record(filename = filename, length = 1))
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
self.test_actions = [TakeMovieAction1(directory = directory,
length = 5,
name = filename,
parameters = "default")]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
os.remove(os.path.join(directory, filename + ".xml"))
self.test_actions = [testActions.SetDirectory(directory = directory),
testActions.ShowGUIControl(control_name = "focus lock"),
testActions.Timer(100),
testActions.Record(filename = filename, length = 10)]
def test_hal_film_7():
# This is expected to record several movies with names starting
# with 'movie_01' and 'movie_02'.
halTest(config_xml="none_classic_config.xml",
class_name="FilmTest6",
test_module="storm_control.test.hal.film_tests")
# Check that the movies are the right length.
for name, size in [["movie_01.dax", [256, 256, 10]],
["movie_01_slice1.dax", [128, 128, 10]],
["movie_02.dax", [256, 256, 10]],
["movie_02_slice1.dax", [128, 128, 10]]]:
movie = datareader.inferReader(os.path.join(test.dataDirectory(),
name))
assert (movie.filmSize() == size)
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_01"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
p_name = "256x256"
self.test_actions = [testActions.LoadParameters(filename = test.halXmlFilePathAndName(p_name + ".xml")),
TakeMovieAction7(directory = directory,
length = 5,
name = filename,
parameters = p_name)]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_02"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
self.test_actions = [testActions.SetDirectory(directory = directory),
testActions.LoadParameters(filename = test.halXmlFilePathAndName("feed_examples.xml")),
testActions.SetParameters(p_name = 0),
testActions.Timer(timeout = 1000),
testActions.Record(filename = filename),
testActions.Timer(timeout = 2000)]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
filename = "movie_03"
# Remove old movie (if any).
fullname = os.path.join(directory, filename + ".dax")
if os.path.exists(fullname):
os.remove(fullname)
os.remove(os.path.join(directory, filename + ".xml"))
self.test_actions = [testActions.SetDirectory(directory = directory),
testActions.Record(filename = filename),
testActions.LoadParameters(os.path.join(directory, "movie_03.xml")),
testActions.SetParameters(p_name = 0),
testActions.Timer(timeout = 500)]
def __init__(self, **kwds):
super().__init__(**kwds)
directory = test.dataDirectory()
self.reps = 1
# Initial setup.
#
# Lock focus.
self.test_actions.append(
tcpTests.SetFocusLockModeAction1(mode_name="Always On",
locked=True))
# Load parameters.
self.test_actions.append(
testActions.LoadParameters(
filename=test.halXmlFilePathAndName("256x256.xml")))
self.test_actions.append(
testActions.LoadParameters(
filename=test.halXmlFilePathAndName("256x512.xml")))
# Turn off live mode.
self.test_actions.append(testActions.SetLiveMode(live_mode=False))
# Pause a second so that setup completes.
self.test_actions.append(testActions.Timer(1000))
# Add loop parameters.
for i in range(10):
## Position 0
# Move stage.
self.test_actions.append(testActionsTCP.MoveStage(x=0.0, y=0.0))
# Check focus.
self.test_actions.append(
testActionsTCP.CheckFocusLock(focus_scan=False,
num_focus_checks=30))
# Set parameters.
self.test_actions.append(
tcpTests.SetParametersAction1(name_or_index="256x256"))
# Take Movie.
self.test_actions.append(
StandardDaveSequenceTakeMovieAction1(directory=directory,
length=5,
name="movie_01"))
# Remove movie.
self.test_actions.append(
testActions.RemoveFile(directory=directory,
name="movie_01.inf"))
self.test_actions.append(
testActions.RemoveFile(directory=directory,
name="movie_01.dax"))
## Position 1
# Move stage.
self.test_actions.append(testActionsTCP.MoveStage(x=10.0, y=0.0))
# Check focus.
self.test_actions.append(
testActionsTCP.CheckFocusLock(focus_scan=False,
num_focus_checks=30))
# Set parameters.
self.test_actions.append(
tcpTests.SetParametersAction1(name_or_index="256x512"))
# Take Movie.
self.test_actions.append(
StandardDaveSequenceTakeMovieAction2(directory=directory,
length=5,
name="movie_02"))
# Remove movie.
self.test_actions.append(
testActions.RemoveFile(directory=directory,
name="movie_02.inf"))
self.test_actions.append(
testActions.RemoveFile(directory=directory,
name="movie_02.dax"))
#!/usr/bin/env python
"""
Test parsing of shutters files.
"""
import numpy
import storm_control.hal4000.illumination.xmlParser as xmlParser
import storm_control.test as test
data_dir = test.dataDirectory()
name_to_id = {"750" : 0,
"647" : 1,
"560" : 2}
def test_parser_1():
"""
Test parsing a correctly formatted XML shutters file.
"""
[s_info, waveforms, oversampling] = xmlParser.parseShuttersXML(name_to_id, data_dir + "shutters_test_1.xml")
# Make we got it right.
assert(s_info.color_data[0] is None)
assert(s_info.color_data[1] == [255,0,0])
assert(s_info.color_data[2] is None)
assert(s_info.frames == 3)
assert(oversampling == 2)
assert(numpy.allclose(numpy.zeros(6), waveforms[0]))
assert(numpy.allclose(numpy.array([0.0, 0.0, 1.0, 1.0, 0.0, 0.0]), waveforms[1]))
assert(numpy.allclose(numpy.zeros(6), waveforms[2]))
#!/usr/bin/env python
"""
Test parsing of shutters files.
"""
import numpy
import storm_control.hal4000.illumination.xmlParser as xmlParser
import storm_control.test as test
data_dir = test.dataDirectory()
name_to_id = {"750": 0, "647": 1, "560": 2}
def test_parser_1():
"""
Test parsing a correctly formatted XML shutters file.
"""
[s_info, waveforms, oversampling
] = xmlParser.parseShuttersXML(name_to_id,
data_dir + "shutters_test_1.xml")
# Make we got it right.
assert (s_info.color_data[0] is None)
assert (s_info.color_data[1] == [255, 0, 0])
assert (s_info.color_data[2] is None)
assert (s_info.frames == 3)
assert (oversampling == 2)
assert (numpy.allclose(numpy.zeros(6), waveforms[0]))
assert (numpy.allclose(numpy.array([0.0, 0.0, 1.0, 1.0, 0.0, 0.0]),