Skip to content

Commit

Permalink
fix one more test
Browse files Browse the repository at this point in the history
  • Loading branch information
guynir42 committed Jun 27, 2024
1 parent 0fe388d commit 0fb9a7f
Show file tree
Hide file tree
Showing 4 changed files with 21 additions and 13 deletions.
3 changes: 1 addition & 2 deletions .github/workflows/run-pipeline-tests-1.yml
Original file line number Diff line number Diff line change
Expand Up @@ -67,5 +67,4 @@ jobs:
sudo rm -rf "$AGENT_TOOLSDIRECTORY"
shopt -s nullglob
# TEST_SUBFOLDER=$(ls tests/pipeline/test_{a..o}*.py) docker compose run runtests
TEST_SUBFOLDER=tests/pipeline/test_making_references.py docker compose run runtests
TEST_SUBFOLDER=$(ls tests/pipeline/test_{a..o}*.py) docker compose run runtests
4 changes: 2 additions & 2 deletions .github/workflows/run-pipeline-tests-2.yml
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,6 @@ jobs:
sudo rm -rf "$AGENT_TOOLSDIRECTORY"
shopt -s nullglob
# TEST_SUBFOLDER=$(ls tests/pipeline/test_{p..z}*.py) docker compose run runtests
TEST_SUBFOLDER=tests/pipeline/test_pipeline.py docker compose run runtests
TEST_SUBFOLDER=$(ls tests/pipeline/test_{p..q}*.py) docker compose run runtests
# TEST_SUBFOLDER=tests/pipeline/test_pipeline.py docker compose run runtests
2 changes: 1 addition & 1 deletion tests/improc/test_sky_flat.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

from util.logger import SCLogger

@pytest.mark.flaky(max_runs=3)
@pytest.mark.flaky(max_runs=6)
@pytest.mark.parametrize("num_images", [10, 300])
def test_simple_sky_flat(num_images):
clear_cache = False # cache the images from the simulator
Expand Down
25 changes: 17 additions & 8 deletions tests/pipeline/test_making_references.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
from pipeline.ref_maker import RefMaker

from models.base import SmartSession
from models.provenance import Provenance
from models.reference import Reference
from models.refset import RefSet

Expand Down Expand Up @@ -64,23 +65,31 @@ def test_finding_references(ptf_ref):


def test_making_refsets():
# make a new refset with a new name
name = uuid.uuid4().hex
maker = RefMaker(maker={'name': name, 'instruments': ['PTF']})
min_number = maker.pars.min_number
max_number = maker.pars.max_number

# we still haven't run the maker, so everything is empty
assert maker.im_provs is None
assert maker.ex_provs is None
assert maker.coadd_im_prov is None
assert maker.coadd_ex_prov is None
assert maker.ref_upstream_hash is None

new_ref = maker.run(ra=0, dec=0, filter='R')
assert new_ref is None

assert new_ref is None # cannot find a specific reference here
refset = maker.ref_set
assert refset is not None

assert refset is not None # can produce a reference set without finding a reference
assert all(isinstance(p, Provenance) for p in maker.im_provs)
assert all(isinstance(p, Provenance) for p in maker.ex_provs)
assert isinstance(maker.coadd_im_prov, Provenance)
assert isinstance(maker.coadd_ex_prov, Provenance)

up_hash1 = refset.upstream_hash
assert maker.ref_upstream_hash == up_hash1
assert isinstance(up_hash1, str)
assert len(up_hash1) == 20
assert len(refset.provenances) == 1
Expand All @@ -91,7 +100,7 @@ def test_making_refsets():

# now make a change to the maker's parameters (not the data production parameters)
maker.pars.min_number = min_number + 5
maker.pars.allow_append = False # this should prevent us from appending to the existing refset
maker.pars.allow_append = False # this should prevent us from appending to the existing ref-set

with pytest.raises(
RuntimeError, match='Found a RefSet with the name .*, but it has a different provenance!'
Expand All @@ -111,15 +120,15 @@ def test_making_refsets():
assert refset.provenances[0].parameters['max_number'] == max_number
assert refset.provenances[1].parameters['max_number'] == max_number

# now try to make a new refset with a different name
# now try to make a new ref-set with a different name
name2 = uuid.uuid4().hex
maker.name = name2
maker.pars.name = name2
new_ref = maker.run(ra=0, dec=0, filter='R')
assert new_ref is None # still can't find images there

refset2 = maker.ref_set
assert len(refset.provenances) == 1
assert refset2.provenances[0].id == refset.provenances[1].id # these refsets share the same provenance!
assert len(refset2.provenances) == 1
assert refset2.provenances[0].id == refset.provenances[1].id # these ref-sets share the same provenance!

# now try to append with different data parameters:
maker.pipeline.extractor.pars['threshold'] = 3.14
Expand Down

0 comments on commit 0fb9a7f

Please sign in to comment.