import chai from 'chai';
import { Matrix, Vector, Sylvester } from '../src';

chai.Assertion.addProperty('matrix', function () {
  new chai.Assertion(this._obj).to.be.instanceOf(Matrix);

  return {
    equal: other => {
      if (!(other instanceof Matrix)) {
        other = Matrix.create(other);
      }

      this.assert(
        this._obj.eql(other),
        'expected #{this} to equal #{exp}',
        'expected #{this} to not equal #{exp}',
        this._obj,
        other,
        true
      );
    }
  };
});

chai.Assertion.addProperty('vector', function () {
  new chai.Assertion(this._obj).to.be.instanceOf(Vector);

  return {
    equal: other => {
      if (!(other instanceof Vector)) {
        other = Vector.create(other);
      }

      this.assert(
        this._obj.eql(other),
        'expected #{this} to equal #{exp}',
        'expected #{this} to not equal #{exp}',
        this._obj,
        other,
        true
      );
    }
  };
});

chai.Assertion.addProperty('approx', function () {
  return {
    equal: other => {
      this.assert(
        Math.abs(this._obj - other) < Sylvester.precision,
        'expected #{this} to about equal #{exp}',
        'expected #{this} to not equal #{exp}',
        this._obj,
        other,
        true
      );
    }
  };
});