Skip to content

Commit

Permalink
improved test
Browse files Browse the repository at this point in the history
  • Loading branch information
jhazentia committed Feb 2, 2024
1 parent b7f7e72 commit 81ec8b6
Showing 1 changed file with 19 additions and 5 deletions.
24 changes: 19 additions & 5 deletions tests/test_sigmffile_collection.py
Expand Up @@ -4,9 +4,23 @@


def test_archive(test_sigmffile, test_alternate_sigmffile, test_alternate_sigmffile_2):
input_sigmffile_collection = SigMFFileCollection(sigmffiles=[test_sigmffile, test_alternate_sigmffile])
input_sigmffile_collection1 = SigMFFileCollection(sigmffiles=[test_sigmffile])
with tempfile.NamedTemporaryFile(suffix=".sigmf") as t_file:
input_sigmffile_collection.archive(fileobj=t_file)
output_sigmf_collection = fromarchive(archive_path=t_file.name)
assert output_sigmf_collection.sigmffile_count() == 2
assert output_sigmf_collection.get_sigmffiles() == [test_sigmffile, test_alternate_sigmffile]
input_sigmffile_collection1.archive(fileobj=t_file)
output_sigmf_collection1 = fromarchive(archive_path=t_file.name)
assert output_sigmf_collection1.sigmffile_count() == 1
assert output_sigmf_collection1.get_sigmffiles() == [test_sigmffile]

input_sigmffile_collection2 = SigMFFileCollection(sigmffiles=[test_sigmffile, test_alternate_sigmffile])
with tempfile.NamedTemporaryFile(suffix=".sigmf") as t_file:
input_sigmffile_collection2.archive(fileobj=t_file)
output_sigmf_collection2 = fromarchive(archive_path=t_file.name)
assert output_sigmf_collection2.sigmffile_count() == 2
assert output_sigmf_collection2.get_sigmffiles() == [test_sigmffile, test_alternate_sigmffile]

input_sigmffile_collection3 = SigMFFileCollection(sigmffiles=[test_sigmffile, test_alternate_sigmffile, test_alternate_sigmffile_2])
with tempfile.NamedTemporaryFile(suffix=".sigmf") as t_file:
input_sigmffile_collection3.archive(fileobj=t_file)
output_sigmf_collection3 = fromarchive(archive_path=t_file.name)
assert output_sigmf_collection3.sigmffile_count() == 3
assert output_sigmf_collection3.get_sigmffiles() == [test_sigmffile, test_alternate_sigmffile, test_alternate_sigmffile_2]

0 comments on commit 81ec8b6

Please sign in to comment.