55import logging
66
77logging .basicConfig (level = logging .DEBUG )
8+ import os
9+ from dotenv import load_dotenv
10+
11+ load_dotenv ()
12+ ENV = os .getenv ("ENV" )
813
914
1015def test_find_biosample_by_id ():
11- biosample = BiosampleSearch ()
16+ biosample = BiosampleSearch (env = ENV )
1217 results = biosample .get_record_by_id ("nmdc:bsm-11-002vgm56" )
1318 assert len (results ) > 0
1419 assert results ["id" ] == "nmdc:bsm-11-002vgm56"
1520
1621
1722def test_logger ():
18- biosample = BiosampleSearch ()
23+ biosample = BiosampleSearch (env = ENV )
1924 logging .basicConfig (level = logging .DEBUG )
2025 results = biosample .get_record_by_id ("nmdc:bsm-11-002vgm56" )
2126
2227
2328def test_biosample_by_filter ():
24- biosample = BiosampleSearch ()
29+ biosample = BiosampleSearch (env = ENV )
2530 results = biosample .get_record_by_filter ('{"id":"nmdc:bsm-11-006pnx90"}' )
2631 assert len (results ) > 0
2732
2833
2934def test_biosample_by_attribute ():
30- biosample = BiosampleSearch ()
35+ biosample = BiosampleSearch (env = ENV )
3136 results = biosample .get_record_by_attribute (
3237 "id" , "nmdc:bsm-11-006pnx90" , exact_match = False
3338 )
@@ -37,23 +42,23 @@ def test_biosample_by_attribute():
3742
3843def test_biosample_by_latitude ():
3944 # {"lat_lon.latitude": {"$gt": 45.0}, "lat_lon.longitude": {"$lt":45}}
40- biosample = BiosampleSearch ()
45+ biosample = BiosampleSearch (env = ENV )
4146 results = biosample .get_record_by_latitude ("gt" , 45.0 )
4247 assert len (results ) > 0
4348 assert results [0 ]["lat_lon" ]["latitude" ] == 63.875088
4449
4550
4651def test_biosample_by_longitude ():
4752 # {"lat_lon.latitude": {"$gt": 45.0}, "lat_lon.longitude": {"$lt":45}}
48- biosample = BiosampleSearch ()
53+ biosample = BiosampleSearch (env = ENV )
4954 results = biosample .get_record_by_longitude ("lt" , 45.0 )
5055 assert len (results ) > 0
5156 assert results [0 ]["lat_lon" ]["longitude" ] == - 149.210438
5257
5358
5459def test_biosample_by_lat_long ():
5560 # {"lat_lon.latitude": {"$gt": 45.0}, "lat_lon.longitude": {"$lt":45}}
56- biosample = BiosampleSearch ()
61+ biosample = BiosampleSearch (env = ENV )
5762 results = biosample .get_record_by_lat_long ("gt" , "lt" , 45.0 , 45.0 )
5863 assert len (results ) > 0
5964 assert results [0 ]["lat_lon" ]["latitude" ] == 63.875088
@@ -62,7 +67,7 @@ def test_biosample_by_lat_long():
6267
6368def test_biosample_build_filter_1 ():
6469 u = DataProcessing ()
65- b = BiosampleSearch ()
70+ b = BiosampleSearch (env = ENV )
6671 filter = u .build_filter ({"name" : "G6R2_NF_20JUN2016" })
6772 results = b .get_record_by_filter (filter )
6873 print (results )
@@ -71,7 +76,7 @@ def test_biosample_build_filter_1():
7176
7277def test_biosample_build_filter_2 ():
7378 u = DataProcessing ()
74- b = BiosampleSearch ()
79+ b = BiosampleSearch (env = ENV )
7580 filter = u .build_filter ({"name" : "G6R2_NF_20JUN2016" , "id" : "nmdc:bsm-11-006pnx90" })
7681 logging .debug ("Biosample test filter:" , filter )
7782 results = b .get_record_by_filter (filter )
0 commit comments