def test_from_single_command(self):
"""Create a story from a starlab command."""
from pystarlab.starlab import Story
king_command = "makeking -w 1.5 -s 1454677882 -n 5 -i"
king_output = "king.out"
king_path = os.path.join(DATA_DIR, king_output)
with open(king_path, 'r') as f:
king_str = f.read()
king_story = Story.from_single_command(king_command)
self.assertEquals(len(king_str), len(str(king_story)))
for line in zip(king_str.splitlines(),
str(king_story).splitlines()):
string_with_date = re.match("^ ===>",line[0])
if not string_with_date:
self.assertEquals(line[0], line[1])
from_list = Story.from_single_command(king_command.split(" "))
for line in zip(king_str.splitlines(),
str(from_list).splitlines()):
string_with_date = re.match("^ ===>",line[0])
if not string_with_date:
self.assertEquals(line[0], line[1])
def test_from_string(self):
"""Test story creation from a string."""
from pystarlab.starlab import Story
king_output = "king.out"
king_path = os.path.join(DATA_DIR, king_output)
with open(king_path, 'r') as f:
king_str = f.read()
king_story = Story.from_string(king_str)
self.assertEquals(king_str, str(king_story))
def test_from_buf(self):
"""Test story creation from an iterable source."""
from pystarlab.starlab import Story
king_output = "king.out"
king_path = os.path.join(DATA_DIR, king_output)
with open(king_path, 'r') as f:
king_str = f.read()
king_story = Story.from_buf(king_str.splitlines())
self.assertEquals(king_str, str(king_story))
def test_command_list(self):
"""Check that we can create a story from a list of commands."""
from pystarlab.starlab import Story
commands = ["makeking -w 1.5 -s 1454677882 -n 5 -i",
"makemass -i -l 0.1 -u 20 -s 1454677882"]
mass_output = "mass.out"
mass_path = os.path.join(DATA_DIR, mass_output)
with open(mass_path, 'r') as f:
mass_str = f.read()
mass_story = Story.from_command_list(commands)
for line in zip(mass_str.splitlines(),
str(mass_story).splitlines()):
string_with_date = re.match("^ ===>",line[0])
if not string_with_date:
self.assertEquals(line[0], line[1])
def test_story_list_from_kira_command(self):
"""Make sure from_buf returns a list of stories when needed.
I'm not trying to test that kira works and I've already
tested that from_buf works correctly for a single story,
so I don't need to test for content here, only the number of
stories returned.
In this case, a run for 10 time units with output every 2 will
give 5 snapshots, not including the initial conditions. Including
initial conditions will bump the number up to 6.
"""
from pystarlab.starlab import Story
commands = ["makeking -w 1.5 -s 1454677882 -n 50 -i",
"makemass -i -l 0.1 -u 20 -s 1454677882",
"kira -t 10 -d 1 -D 2"]
kira_snaps = Story.from_command_list(commands)
self.assertEquals(len(kira_snaps), 6)
def test_apply_command(self):
"""Test chaining a command onto an story, using it as input."""
from pystarlab.starlab import Story
king_command = "makeking -w 1.5 -s 1454677882 -n 5 -i"
mass_command = "makemass -i -l 0.1 -u 20 -s 1454677882"
mass_output = "mass.out"
mass_path = os.path.join(DATA_DIR, mass_output)
with open(mass_path, 'r') as f:
mass_str = f.read()
king_story = Story.from_single_command(king_command)
mass_story = king_story.apply_command(mass_command)
for line in zip(mass_str.splitlines(),
str(mass_story).splitlines()):
string_with_date = re.match("^ ===>",line[0])
if not string_with_date:
self.assertEquals(line[0], line[1])