def test_01(self):
cs = CertificateSearch()
cs.set_period(2013, 1)
cs.filter_technology("Wind")
cs.filter_scheme('REGO')
self.assertTrue(cs.get_data())
self.assertIsInstance(cs[0], Certificates)
self.assertEqual(cs[0].period, 'Jan-2013')
print("Contacting Ofgem and preparing to search.\n")
ocs = CertificateSearch()
crit = "Searching Ofgem Certificates: "
crits = []
if args.scheme:
ocs.filter_scheme(args.scheme)
crits.append('\n\tscheme %s' % args.scheme)
if args.generator:
ocs.filter_generator_id(args.generator.upper())
crits.append("\n\tgenerator id is '%s'" % args.generator.upper())
if args.month and args.year:
ocs.set_period(args.year, args.month)
crits.append('\n\tperiod should be {} {}'.format(args.month, args.year))
else:
if args.month:
ocs.set_month(args.month)
crits.append('\n\tmonth %s' % args.month)
if args.year:
ocs.set_year(args.year)
crits.append('\n\tyear %s' % args.year)
print("Searching Ofgem for certificates matching:{}\n".format(", ".join(crits)))
ocs.get_data()
else:
ocs = CertificateSearch(filename=args.filename)
print("Total of %d records returned" % len(ocs))
