def test_3_getStarCount(self): s = StarCatalog() s.loadCatalog("star_file_test.txt") try: s.getStarCount("hi", 1) except: print ("Test 3.1 Passed: Invalid LowerMagnitude") try: s.getStarCount(1, "lol") except: print ("Test 3.2 Passed: Invalid UpperMagnitude") try: s.getStarCount(30, 1) except: print ("Test 3.3 Passed: Lower Magnitude > Upper Magnitude") if (s.getStarCount() == 3): print ("Test 3.4 Passed: No params, returned all stars") try: count = s.getStarCount(0, 5) except: print("Test 3.5 failed!!!!") if (count == 3): print ("Test 3.5 passed: Normal getStarCount") else: print("Test 3.5 failed!!!!")
def test_5_loadCatalog(self): s = StarCatalog() s.loadCatalog("star_file_test.txt") if (s.getStarCount(None, None) == 2): print"Test 5.1 passed: loaded 2 stars"